np.delete删除若干行
时间: 2023-08-03 11:08:33 浏览: 78
np.delete函数可以用来删除数组中的元素,其中可以指定删除的维度和索引。如果要删除若干行,可以指定删除的维度为0,即数组的行维度。以下是一个示例代码:
```python
import numpy as np
# 创建一个5x5的二维数组
arr = np.arange(25).reshape((5, 5))
print("原始数组:")
print(arr)
# 删除第2行和第4行
arr = np.delete(arr, [1, 3], axis=0)
print("删除后的数组:")
print(arr)
```
输出结果如下:
```
原始数组:
[[ 0 1 2 3 4]
[ 5 6 7 8 9]
[10 11 12 13 14]
[15 16 17 18 19]
[20 21 22 23 24]]
删除后的数组:
[[ 0 1 2 3 4]
[10 11 12 13 14]
[20 21 22 23 24]]
```
在这个例子中,我们首先创建了一个5x5的二维数组,然后使用np.delete函数删除了第2行和第4行,最后打印出了删除后的数组。在np.delete函数中,第一个参数是要删除的数组,第二个参数是要删除的索引,第三个参数是要删除的维度。因为我们要删除行,所以指定维度为0。
阅读全文