重复的numpy数组行,其中重复的数量不同

时间:2013-09-11 06:48:12

标签: python numpy

虽然有一个numpy数组,人们希望将每个值复制指定的次数:

np.array([1,2,3,4])

和第二个数组定义原始数组中每个相应索引位置所需的重复数:

np.array([3,3,2,2])

如何生产:

[1,1,1,2,2,2,3,3,4,4]

显然,可以使用迭代来生成新数组,但我很好奇是否有更优雅的基于numpy的解决方案。

1 个答案:

答案 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])