高效插入数字numpython

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