如何在 numpy 中向矩阵添加列 [重复]

Question

我想在 0 索引处将一列添加到 5x5 矩阵。 np.append() 正在工作，但我搜索另一种方式。

import numpy as np
arr = np.array(range(25)).reshape(5,5)
ones = np.ones((arr.shape[0],1))
arr_with_ones = np.append(ones, arr, axis=1)
print(arr_with_ones)

Answer 1

您无需预定义ones 数组。可以直接使用numpy.insert函数：

arr = np.array(range(25)).reshape(5,5)
arr_with_ones = np.insert(arr, 0, 1, axis=1)

np.insert(arr, 0, 1, axis=1) 在数组arr 的axis=1（二维数组中的列）的索引0 中插入value=1。

输出：

[[ 1  0  1  2  3  4]
 [ 1  5  6  7  8  9]
 [ 1 10 11 12 13 14]
 [ 1 15 16 17 18 19]
 [ 1 20 21 22 23 24]]

Answer 2

你可以这样试试：

import numpy as np
arr = np.array(range(25)).reshape(5,5)
ones = np.ones((arr.shape[0],1))
arr_with_ones = np.c_[ones, arr]

Answer 3

您也可以尝试np.hstack((ones,arr)) 功能。