Python+numpy实现矩阵的行列扩展

当前位置:

嘿，程序员小伙伴！在Python里用NumPy来扩展矩阵的行列可是个基本操作，我来给你展示一下怎么做吧。

首先，你需要确保已经安装了NumPy库。如果还没有安装，可以通过以下命令进行安装：

pip install numpy

安装好了之后，我们就可以开始扩展矩阵的行列了。这里有两种常见的方式：扩展行和扩展列。

### 1. 扩展行

假设我们有一个2x3的矩阵：

import numpy as np

# 创建一个2x3的矩阵
matrix = np.array([[1, 2, 3],
[4, 5, 6]])

现在，我们想要扩展这个矩阵的行，比如在末尾添加一行 `[7, 8, 9]`。

# 使用np.vstack进行垂直堆叠（扩展行）
new_row = np.array([[7, 8, 9]])
expanded_matrix = np.vstack((matrix, new_row))

print(expanded_matrix)

输出：

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

### 2. 扩展列

同样的，假设我们有一个2x3的矩阵：

import numpy as np

# 创建一个2x3的矩阵
matrix = np.array([[1, 2, 3],
[4, 5, 6]])

现在，我们想要扩展这个矩阵的列，比如在末尾添加一列 `[7, 8]`。

# 使用np.hstack进行水平堆叠（扩展列）
new_column = np.array([[7], [8]])
expanded_matrix = np.hstack((matrix, new_column))

print(expanded_matrix)

输出：

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

### 小贴士

- `np.vstack` 用于垂直堆叠（即扩展行），输入需要是形状为 `(n_rows, *)` 的数组序列。
- `np.hstack` 用于水平堆叠（即扩展列），输入需要是形状为 `(*, n_cols)` 的数组序列。

希望这个解释能帮到你，如果有更多问题，尽管来问我！

最后，如果你对python语言还有任何疑问或者需要进一步的帮助，请访问https://www.xin3721.com 本站原创，转载请注明出处：https://www.xin3721.com/Python/python50584.html

栏目列表