numpy删除全零行
时间: 2024-08-21 10:00:43 浏览: 51
Numpy(Pandas)删除全为零的列的方法
在NumPy库中,如果你想要删除数组中的所有全零行(即行元素全部为0),你可以使用`numpy.where()`函数配合布尔索引来找出非零值所在的行,然后通过切片操作来剔除这些零行。这里有一个简单的例子:
```python
import numpy as np
# 假设你有一个二维数组 arr
arr = np.array([[0, 0, 0], [1, 2, 3], [0, 4, 5], [0, 0, 6]])
# 使用 where() 函数找到非零元素所在的位置(行索引)
non_zero_rows = np.where(arr != 0)
# 删除全零行
filtered_arr = arr[non_zero_rows]
print(filtered_arr)
```
运行上述代码后,`filtered_arr` 将只包含非全零行:[[1, 2, 3], [4, 5]]。
阅读全文