删除np.array中的某行数据
时间: 2023-09-17 18:13:34 浏览: 120
如何实现删除numpy.array中的行或列
可以使用 NumPy 的 delete() 函数来删除数组中的某行数据。delete() 函数的语法如下:
```python
numpy.delete(arr, obj, axis=None)
```
其中,arr 表示要删除元素的数组,obj 表示要删除的元素下标或下标数组,axis 表示删除元素的轴方向。当 axis 为 None 时,表示删除 flatten 后的数组中的元素;当 axis 为 0 时,表示删除 arr 的第一维元素;当 axis 为 1 时,表示删除 arr 的第二维元素,以此类推。
例如,要删除 arr 数组的第 2 行,可以使用如下代码:
```python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
new_arr = np.delete(arr, 1, axis=0)
print(new_arr)
```
输出结果为:
```
[[1 2 3]
[7 8 9]]
```
其中,np.delete(arr, 1, axis=0) 表示删除 arr 数组的第 2 行(下标为 1),axis=0 表示删除第一维元素,即删除行。
阅读全文