Skip to main content
 首页 » 编程设计

python之从索引位置列表中创建非均匀(随机)间隔的范围

2025年05月04日12oomusou

假设我有一个数组和一个位置列表:

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()