python - 展开一维 numpy 数组

我想扩展/重复一维数组中的每个元素不同的时间

x = array([7,1,9])  #1d array
rep_size = array([3,2,2]) # repeat number for each element in x
result = arary([7,7,7,1,1,9,9]) #expected result

如果我不想使用 for 循环,是否有 numpy 函数可以执行此操作。谢谢。

回答1

使用 https://numpy.org/doc/stable/reference/generated/numpy.repeat.html

result = np.repeat(x, rep_size)

输出:array([7, 7, 7, 1, 1, 9, 9])

相似文章