虽然有一个numpy数组,人们希望将每个值复制指定的次数:
np.array([1,2,3,4])
和第二个数组定义原始数组中每个相应索引位置所需的重复数:
np.array([3,3,2,2])
如何生产:
[1,1,1,2,2,2,3,3,4,4]
显然,可以使用迭代来生成新数组,但我很好奇是否有更优雅的基于numpy的解决方案。
答案 0 :(得分:3)
使用numpy.repeat
:
>>> numpy.repeat([1,2,3,4], [3,3,2,2])
array([1, 1, 1, 2, 2, 2, 3, 3, 4, 4])