在numpy数组

时间:2017-11-06 09:41:49

标签: python-3.x numpy

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()方法允许在输入元素之间仅添加一个数字。让我知道你对此的看法。

1 个答案:

答案 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])