展开一维 numpy 数组
expand 1d numpy array
我想expand/repeat一维数组中的每个元素在不同的时间
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 函数可以执行此操作。谢谢
使用numpy.repeat:
result = np.repeat(x, rep_size)
输出：array([7, 7, 7, 1, 1, 9, 9])
我想expand/repeat一维数组中的每个元素在不同的时间
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 函数可以执行此操作。谢谢
使用numpy.repeat:
result = np.repeat(x, rep_size)
输出：array([7, 7, 7, 1, 1, 9, 9])