假设我有一个数组和一个位置列表:
arr = np.ones(35)
[3, 5, 8, 12, 14, 17, 19, 25, 27, 33]
在这些不同的点,我想增加一个,这样我就有了一个类似于
的最终数组array([1,1,1,2,2,3,3,3,4,4,4,4,5,5,6,6,6,7,7,8,8,....])
请您参考如下方法:
arr = np.zeros(35, dtype=int)
idx = [3, 5, 8, 12, 14, 17, 19, 25, 27, 33]
arr[0] = 1
arr[idx] = 1
result = arr.cumsum()
