python - 长pandas逻辑运算

Question

有没有更好的方法来编写这么长的 pandas 逻辑操作？下面我只列出了几个例子，但实际上还有更多的检查。这可以完成工作，但我觉得可能有一种更清晰的方式来表达这个操作。如果您对此数据感到好奇，它是 NYC taxi data如有任何建议，我们将不胜感激。

mask = (df.Fare_amount >= 2.5) & \
       (df.Total_amount >= 2.5) & \
       (df.Tip_amount >= 0) & \
       (df.Tolls_amount >= 0) & \
       (df.Extra >= 0) & \
       (df.Trip_distance > 0) & \
       (df.Passenger_count.between(1, 5))

# I actually have some more checks here like these above....

new_df = df[mask].copy()

Answer 1

不确定是否有更好的方法，但一种选择可能是将具有相同条件的列分组在一起，并在行上使用 all 来减少 & 的数量> 使用的操作:

mask = (df[['Fare_amount', 'Total_amount']] >= 2.5).all(1) & \
       (df[['Tip_amount', 'Tolls_amount', 'Extra']] >= 0).all(1) & \
       (df.Trip_distance > 0) & \
       (df.Passenger_count.between(1, 5))