📅  最后修改于: 2023-12-03 15:09:08.021000             🧑  作者: Mango
在 Python 中使用 numpy 库时,经常需要在现有的矩阵中添加新的行或列。本篇文章将介绍如何将行添加到 numpy 矩阵中。
使用 np.vstack() 方法将行添加到一个 numpy 矩阵中。该方法将原矩阵和要添加的行作为参数,返回一个新的矩阵。
import numpy as np
# 定义一个 2x2 的矩阵
a = np.array([[1, 2], [3, 4]])
# 定义要添加的行
row_to_append = np.array([5, 6])
# 使用 np.vstack() 方法将行添加到矩阵中
new_matrix = np.vstack((a, row_to_append))
# 打印新矩阵
print(new_matrix)
输出结果:
[[1 2]
[3 4]
[5 6]]
np.append() 方法可以将多个数组沿着一个指定的轴连接在一起,并返回连接后的新数组。在添加行时,需要将要添加的行作为一个新的数组和轴参数传递给 np.append() 方法。
import numpy as np
# 定义一个 2x2 的矩阵
a = np.array([[1, 2], [3, 4]])
# 定义要添加的行
row_to_append = np.array([5, 6]).reshape(1,2)
# 使用 np.append() 方法将行添加到矩阵中
new_matrix = np.append(a, row_to_append, axis=0)
# 打印新矩阵
print(new_matrix)
输出结果:
[[1 2]
[3 4]
[5 6]]
本篇文章介绍了两种在 numpy 矩阵中添加行的方法:使用 np.vstack() 方法和 np.append() 方法。在实际应用中,可以根据具体需求选择不同的方法。