我想擴展/重復一維陣列中的每個元素不同的時間
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 函式可以做到這一點。謝謝。
uj5u.com熱心網友回復:
使用numpy.repeat:
result = np.repeat(x, rep_size)
輸出:array([7, 7, 7, 1, 1, 9, 9])
轉載請註明出處,本文鏈接:https://www.uj5u.com/qianduan/477483.html
