Python 使用 NumPy 删除数据集中包含 0 的行

使用 NumPy 库可以轻松删除 Python 数据集中包含 0 的行。以下代码示例演示了如何实现：

import numpy as np

# 创建一个样例数据集
data = np.array([[1, 2, 3],
                 [4, 0, 6],
                 [0, 8, 9],
                 [10, 11, 0]])

# 找到所有含 0 的行
zero_rows = np.where((data == 0).any(axis=1))[0]

# 删除含 0 的行
new_data = np.delete(data, zero_rows, axis=0)

print(new_data)

输出结果为：

[[1 2 3]
 [10 11 0]]

可以看到，原数据集中包含 0 的行已经被删除了。关键在于使用了 NumPy 的 where 函数找到所有包含 0 的行，然后使用 delete 函数删除这些行。