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()方法允许在输入元素之间仅添加一个数字。让我知道你对此的看法。
答案 0 :(得分:0)
您可以使用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])