如何重複或平鋪numpy數組，但每次平鋪時都要更改一個元素中的值？

Question

比方說有一個numpy的陣列 a = [1,1,1,0]

我想瓷磚或重複此陣3次，但每次1是平鋪/重複時間使最後一個元素增加。

也就是說，我想

result = [[1,1,1,0], [1,1,1,1], [1,1,1,2]] 

到底

。

我想我看到有人用一個函數來做到這一點，但我不記得那個函數是什麼。 或者我可能是錯的。

Answer 1

import numpy as np 

a = np.array([1, 1, 1, 0]) 

#how often to repeat the array along first dimension? 
b = 20 

#repeat b times along first dimension, one time along second 
x = np.tile(a, (b,1)) 

print(x) 

#just some consecutive numbers 
y = np.arange(20) 
print(y) 

#overwrite fourth column of array 
x[:, 3] = y 

print(x)