python - 将一列添加到 numpy 数组

Question

我试图简单地将一列添加到 numpy 数组中，但找不到任何简单的解决方案来解决我认为应该是一个简单的答案。我的数组中的行数可能会发生变化，因此解决方案需要概括。

import numpy as np
X = np.array([[1,45,23,56,34,23], 
              [2,46,24,57,35,23]])

我想要的输出:

array([[ 1, 45, 23, 56, 34, 23, 1],
       [ 2, 46, 24, 57, 35, 23, 1]])

我尝试过使用 np.append 和 np.insert，但它们要么压平数组，要么替换值。

谢谢。

Answer 1

你可以做hstack:

np.hstack((X,np.ones([X.shape[0],1], X.dtype)))

输出:

array([[ 1, 45, 23, 56, 34, 23,  1],
       [ 2, 46, 24, 57, 35, 23,  1]])