📅  最后修改于: 2023-12-03 14:52:09.245000             🧑  作者: Mango
删除 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 数组。