重复numpy数组行，其中重复次数不同

Question

当需要复制numpy数组中的每个值指定次数时：

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

另一个数组定义了原始数组中每个相应索引位置所需的重复次数：

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

如何生成：

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

显然，可以使用迭代来生成新的数组，但我想知道是否有更优雅的基于numpy的解决方案。

- metasequoia

Answer 1

使用 numpy.repeat:

>>> numpy.repeat([1,2,3,4], [3,3,2,2])
array([1, 1, 1, 2, 2, 2, 3, 3, 4, 4])

- user2357112