Python如何移除NumPy数组中包含零值的行

• 不需要写多层循环，直接通过NumPy的布尔索引即可实现逐行筛选，效率远高于遍历判断
• 针对你给出的严格零值场景，可直接用如下代码完成筛选：

import numpy as np

# 复现你的示例数组用于验证
AA = np.array([[2.40258090e+01, 5.46388441e+00],
       [2.37367092e+01, 1.63425169e+02],
       [0.00000000e+00, 1.56246202e-01],
       [1.68035590e+01, 0.00000000e+00],
       [1.06032437e+01, 6.41505749e+00]])

# 逐行判断：保留所有行内元素全部不为0的行
filtered_AA = AA[np.all(AA != 0, axis=1)]

运行这段代码得到的filtered_AA会自动剔除示例中第三行、第四行存在零值的行，和预期结果完全一致。

• 适配浮点精度场景：如果你的数组经过计算存在浮点误差，所谓的“零值”并非严格等于0.0，可把非零判断替换为零值临近判断，避免漏筛：

filtered_AA = AA[np.all(~np.isclose(AA, 0), axis=1)]

• 调试失败的常见原因：绝大多数人尝试这类逻辑时结果不符合预期，都是漏写了 axis=1 参数。如果不指定按行维度聚合，np.all()会对整个数组做全局判断——只要全数组存在任意一个零值就返回False，最终得到的筛选结果和逐行过滤的需求完全不匹配。

内容的提问来源于stack exchange，提问作者user220318