在numpy数组

Question

python中是否有一个函数允许我在数组[1,2,3,4,5]中插入数字100或连续非零？

输出应为[1,100,100,100,2,100,100,100 ......]或[1,100,101,102,2,100,101,102,3。 ..]

我试过numpy.insert（）

ar2 = np.insert（ar1，slice（1，None），range（100,103））

输出：数组（[1,100,2,101,3,102,4,100,5,101]）

Numpy.Insert（）方法允许在输入元素之间仅添加一个数字。让我知道你对此的看法。

Answer 1

您可以使用numpy.kron

np.kron([1,2,3,4,5],[1,0,0,0]) + 100*np.kron(np.ones(5),[0,1,1,1])

为第二个

np.kron([1,2,3,4,5],[1,0,0,0]) + np.kron(np.ones(5),[0,101,102,103])