python – Pandas根据多个条件过滤行

前端之家收集整理的这篇文章主要介绍了python – Pandas根据多个条件过滤行前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

我使用pandas 0.13.1 Python 2.7:

我在风险列中有一些既不是小,中,高的值.我想删除值不是小,中和高的行.我尝试了以下方法

  1. df = df[(df.risk == "Small") | (df.risk == "Medium") | (df.risk == "High")]

但是这会返回一个空数据框.我该如何正确过滤它们?

最佳答案
我想你想要:

  1. df = df[(df.risk.isin(["Small","Medium","High"]))]

例:

  1. In [5]:
  2. import pandas as pd
  3. df = pd.DataFrame({'risk':['Small','High','Medium','Negligible','Very High']})
  4. df
  5. Out[5]:
  6. risk
  7. 0 Small
  8. 1 High
  9. 2 Medium
  10. 3 Negligible
  11. 4 Very High
  12. [5 rows x 1 columns]
  13. In [6]:
  14. df[df.risk.isin(['Small','High'])]
  15. Out[6]:
  16. risk
  17. 0 Small
  18. 1 High
  19. 2 Medium
  20. [3 rows x 1 columns]

猜你在找的Python相关文章