python - Numpy 数组 : concatenate arrays and integers

Question

在我的 Python 程序中，我连接了几个整数和一个数组。如果这可行的话，那就很直观了:

x,y,z = 1,2,np.array([3,3,3])
np.concatenate((x,y,z))

但是，所有整数都必须转换为 np.arrays:

x,y,z = 1,2,np.array([3,3,3])
np.concatenate((np.array([x]),np.array([y]),z))

特别是如果您有很多变量，则手动转换会很乏味。问题在于 x 和 y 是 0 维数组，而 z 是 1 维数组。有什么方法可以在不进行转换的情况下进行串联吗？

Answer 1

它们只需是序列对象，不一定是 numpy 数组:

x,y,z = 1,2,np.array([3,3,3])
np.concatenate(([x],[y],z))
# array([1, 2, 3, 4, 5])

Numpy 也有一个 insert 函数可以执行此操作:

x,y,z = 1,2,np.array([3,3,3])
np.insert(z, [0,0], [x, y])

我要补充一点，如果你只是想将整数添加到列表中，则不需要 numpy 来执行此操作:

x,y,z = 1,2,[3,3,3]
z = [x] + [y] + z

或

x,y,z = 1,2,[3,3,3]
[x, y] + z

或

x,y,z = 1,2,[3,3,3]
z.insert(0, y)
z.insert(0, x)