高效地插入數字 - NumPy/Python

Question

返回一個新的向量，其中每個元素由4個連續的零分隔。試圖實現[4,2,1] - > [4,0,0,0,0,2,0,0,0,0,1]

def zero_insert(x): 
    y = np.zeros((5*(len(x)-1))+1, dtype=np.int) 
    for i in range(len(x)): 
     y[5*i] = x[i] 
    return y 

Answer 1

初始化和分配 -

x = np.asarray(x) # convert to array 
n = 4    # number of zeros to be inserted 
N = n+1 
out = np.zeros((len(x)-1)*N+1,dtype=x.dtype) 
out[::N] = x