在特定位置向 numpy 数组插入新行 - Python

在使用 numpy 库时，有时需要向已有的数组中插入新行。下面给出两种方法：

方法一：使用 numpy.insert() 函数

可以使用 numpy.insert() 函数在特定位置插入新行。下面是一个示例代码：

import numpy as np

# 创建一个 3x3 的数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 在第二行后插入新行
new_row = np.array([10, 11, 12])
arr = np.insert(arr, 2, new_row, axis=0)

# 打印插入新行后的数组
print(arr)

输出结果为：

array([[ 1,  2,  3],
       [ 4,  5,  6],
       [10, 11, 12],
       [ 7,  8,  9]])

在上面的代码中，我们首先创建了一个 3x3 的数组 arr，然后使用 numpy.insert() 函数在第二行后插入了新行。其中，第一个参数为原数组，第二个参数为插入新行的位置，也就是第三行，第三个参数为要插入的新行的值，最后一个参数为数组的维度。

方法二：使用 numpy.concatenate() 函数

还可以使用 numpy.concatenate() 函数在特定位置插入新行。下面是一个示例代码：

import numpy as np

# 创建一个 3x3 的数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 在第二行后插入新行
new_row = np.array([10, 11, 12])
arr = np.concatenate((arr[:2,:], [new_row], arr[2:,:]))

# 打印插入新行后的数组
print(arr)

输出结果为：

array([[ 1,  2,  3],
       [ 4,  5,  6],
       [10, 11, 12],
       [ 7,  8,  9]])

在上面的代码中，我们同样创建了一个 3x3 的数组 arr，然后使用 numpy.concatenate() 函数在第二行插入了新行。其中，第一个参数为一个元组，包含三部分：第一部分为原数组的前两行（即第一行和第二行），第二部分为要插入的新行，第三部分为原数组的第三行。最后得到的数组就是插入新行后的数组。

总的来说，这两种方法都可以实现向 numpy 数组中插入新行的功能。具体使用哪一种方法取决于实际情况，需要结合具体问题进行选择。