將一列添加到 numpy 數組

Question

我試圖簡單地將一列添加到 numpy 數組中，但找不到任何簡單的解決方案來解決我認為應該是一個簡單的答案。 我的數組中的行數可能會改變，因此解決方案需要概括。

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

我想要的 output：

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)))

Output：

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

Answer 2

您可以使用append ，但您必須告訴它您希望它沿哪個axis工作：

np.append(X, [[1],[1]],  axis=1)  

Answer 3

您可以使用numpy.c_

np.c_[X, [1, 1]]

Answer 4

您可以使用numpy.insert以下方式：

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

Output：

[[ 1 45 23 56 34 23  1]
 [ 2 46 24 57 35 23  1]]