pandas:如果多列之一以上为零,则删除行

pandas:如果多列之一以上为零,则删除行,第1张

pandas:如果多列之一以上为零,则删除

应用条件并计算

True
值。

(df == 0).sum(1)ID1    2ID2    0ID3    1dtype: int64df[(df == 0).sum(1) < 2]     col0  col1  col2  col3ID2     1     1     2    10ID3     0     1     3     4

或者,将整数转换为bool并将其求和。更直接一点。

# df[(~df.astype(bool)).sum(1) < 2]df[df.astype(bool).sum(1) > len(df.columns)-2]  # no inversion needed     col0  col1  col2  col3ID2     1     1     2    10ID3     0     1     3     4

为了提高性能,您可以使用

np.count_nonzero

# df[np.count_nonzero(df, axis=1) > len(df.columns)-2]df[np.count_nonzero(df.values, axis=1) > len(df.columns)-2]     col0  col1  col2  col3ID2     1     1     2    10ID3     0     1     3     4

df = pd.concat([df] * 10000, ignore_index=True)%timeit df[(df == 0).sum(1) < 2]%timeit df[df.astype(bool).sum(1) > len(df.columns)-2]%timeit df[np.count_nonzero(df.values, axis=1) > len(df.columns)-2]7.13 ms ± 161 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)4.28 ms ± 120 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)997 µs ± 38.2 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)


欢迎分享,转载请注明来源:内存溢出

原文地址: https://outofmemory.cn/zaji/5630874.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存