向 numpy 数组添加新行 - Python

在使用 numpy 时，有时需要在一个数组中添加新的行。下面将介绍两种方法来实现这个目的。

方法一：使用 vstack()

可以使用 numpy 的 vstack() 函数在数组底部添加新行。下面是一个简单的示例：

import numpy as np

# 创建一个2维数组
arr = np.array([[1, 2, 3],
                [4, 5, 6]])

# 创建新的行
new_row = np.array([7, 8, 9])

# 在底部添加新的行
new_arr = np.vstack((arr, new_row))

print(new_arr)

输出：

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

在上面的代码中，我们首先创建了一个 arr 数组，然后定义了一个新的行 new_row。接下来，我们使用 vstack() 函数将 new_row 添加到 arr 的底部，生成新的数组 new_arr。

方法二：使用 append()

另一种方法是使用 numpy 的 append() 函数。下面是一个示例：

import numpy as np

# 创建一个2维数组
arr = np.array([[1, 2, 3],
                [4, 5, 6]])

# 创建新的行
new_row = np.array([7, 8, 9])

# 在底部添加新的行
new_arr = np.append(arr, [new_row], axis=0)

print(new_arr)

输出：

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

在上面的代码中，我们首先创建了一个 arr 数组，然后定义了一个新的行 new_row。接下来，我们使用 append() 函数将 new_row 添加到 arr 的底部，生成新的数组 new_arr。注意，我们需要明确指定 axis=0，以便将 new_row 添加到 arr 的底部。