Python编程给numpy矩阵添加一列方法示例

  • Post category:Python

以下是关于“Python编程给numpy矩阵添加一列方法示例”的完整攻略。

给numpy矩阵添加一列

在Python中,可以使用numpy库中的concatenate()函数和reshape()函数来给numpy矩阵添加一列。具体步骤如下:

  1. 创建一个新的一维数组,作为要添加的列;
  2. 使用concatenate()函数将原矩阵和新数组按列连接;
  3. 使用reshape()函数将连接后的数组重新变为矩阵。

示例1:给numpy矩阵添加一列

假设我们有一个numpy矩阵a,如下所示:

import numpy as np

a = np.array([[1, 2], [3, 4], [5, 6]])

我们可以使用concatenate()函数和reshape()函数来给矩阵a添加一列,示例代码如下:

b = np.array([7, 8, 9])
c = np.concatenate((a, b.reshape(-1, 1)), axis=1)
print(c)

在上面的示例代码中,我们首先创建了一个新的一维数组b,作为要添加的列。然后,我们使用concatenate()函数将原矩阵a和新数组b按列连接,并将结果存储在数组c中。最后,我们输出了连接后的矩阵c

输出结果如下:

[[1 2 7]
 [3 4 8]
 [5 6 9]]

示例2:给numpy矩阵添加多列

假设我们有一个numpy矩阵a,如下所示:

import as np

a = np.array([[1, 2], [3, 4], [5, 6]])

我们可以使用concatenate()函数和reshape()函数来给矩阵a添加多列,示例代码如下:

b = np.array([[7, 8, 9], [10, 11, 12], [13, 14, 15]])
c = np.concatenate((a, b), axis=1)
print(c)

在上面的示例代码中,我们首先创建了一个新的二维数组b,作为要添加的多列。然后,我们使用concatenate()函数将原矩阵a和新数组b按列连接,并将结果存储在数组c中。最后,我们输出了连接后的矩阵c

输出结果如下:

[[ 1  2  7  8  9]
 [ 3  4 10 11 12]
 [ 5  6 13 14 15]]

总结

综上所述,“Python编程给numpy矩阵添加一列方法示例”的整个攻略包括了给numpy矩阵添加一列的步骤和两个示例。在实际应用中,可以根据具体需求使用concatenate()函数和reshape()函数来给numpy矩阵添加一列或多列。