python - 重复 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 的解决方案。

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