numpy.ndarray中删除多行数据
时间: 2024-05-04 22:19:01 浏览: 102
如何实现删除numpy.array中的行或列
5星 · 资源好评率100%
可以使用`numpy.delete()`函数来删除`ndarray`中的多行数据。该函数的用法如下:
```
numpy.delete(arr, obj, axis=None)
```
其中`arr`是要删除数据的`ndarray`,`obj`是要删除的行的索引,`axis`指定了要在哪个轴上删除数据,默认为`None`,表示展开整个数组进行删除。
例如,如果要删除`ndarray`中第1行和第3行,可以这样写:
```python
import numpy as np
a = np.array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9],
[10, 11, 12]])
a = np.delete(a, [0, 2], axis=0)
print(a)
```
输出结果为:
```
[[ 4 5 6]
[10 11 12]]
```
上述代码中,`a`是一个4行3列的`ndarray`,使用`np.delete()`函数删除了第1行和第3行。
阅读全文