numpy - 如何重复或平铺一个 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)