📜  如何删除多行 NumPy 数组?(1)

📅  最后修改于: 2023-12-03 14:52:09.245000             🧑  作者: Mango

如何删除多行 NumPy 数组?

删除 NumPy 数组中的多行,可以使用 numpy.delete() 函数。该函数的语法如下:

numpy.delete(arr, obj, axis=None)

其中,arr 为要删除元素的数组;obj 表示要删除元素的下标或下标数组;axis 表示要删除元素所在的轴,如果未指定,则默认为 None,即在展开的数组中查找元素。

下面我们通过一个示例来演示如何使用 numpy.delete() 函数删除多行 NumPy 数组。

首先,我们创建一个 3 行 3 列的 NumPy 数组:

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

print(arr)

输出结果如下:

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

接下来,我们使用 numpy.delete() 函数删除第一行和第二行:

new_arr = np.delete(arr, [0, 1], axis=0)

print(new_arr)

输出结果如下:

array([[7, 8, 9]])

可以看到,第一行和第二行已经被删除,只剩下第三行。

需要注意的是,numpy.delete() 函数返回的是一个新的数组,原始数组并没有改变。

另外,如果要删除多列,可以将 axis 参数设为 1,然后指定要删除的列的下标或下标数组即可。

到此为止,我们已经介绍了如何删除多行 NumPy 数组。