我想扩展/重复一维数组中的每个元素不同的时间
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])