展开一维 numpy 数组

Question

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

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 函数可以做到这一点。 谢谢。

Answer 1

使用numpy.repeat ：

result = np.repeat(x, rep_size)

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