python - 重复 numpy 数组行,其中重复次数不同

标签 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 的解决方案。

最佳答案

使用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/

相关文章:

python - Scrapy 异常 - exceptions.AttributeError : 'unicode' object has no attribute 'select'

python - 在 Tkinter 中回调字体

python - 从 numpy 中的图像中提取缩略图

python - pandas 对 SparseSeries 列表的内存使用情况

python - 尝试使用 python(matplotlib 和 numpy)绘制函数,如果我稍微改变它,就会遇到值错误和类型错误

python - Numpy Pandas 安装失败,/dev/null 上存在沙箱冲突

python - 从 Python 中的列表列表中删除重复项

python - "save() got an unexpected keyword argument ' 在功能 View 中提交 '"错误

python - 按一组 Pandas 分组

python - 通过 numpy 坐标数组索引 numpy 数组