鉴于有一个 numpy 数组,人们希望将每个值复制指定的次数:
np.array([1,2,3,4])
第二个数组定义原始数组中每个相应索引位置所需的重复次数:
np.array([3,3,2,2])
如何生产:
[1,1,1,2,2,2,3,3,4,4]
显然,可以使用迭代来生成新数组,但我很好奇是否有更优雅的基于 numpy 的解决方案。
最佳答案
使用numpy.repeat
:
>>> numpy.repeat([1,2,3,4], [3,3,2,2])
array([1, 1, 1, 2, 2, 2, 3, 3, 4, 4])
关于python - 重复 numpy 数组行,其中重复次数不同,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/18734447/