如何使用字符串列表作为datafram中的条件

Question

如果我有这样一个数据帧（df）：

Date Temperature Climate
4/1  50          Sunny
4/2  55          Cloudy
4/3  48          Rainy
4/4  53          Windy
4/5  33          Snowy
...

我想挑选我感兴趣的天气

climate_of_interest = ['Sunny','Rainy','Snowy']

我如何设置一个条件来只得到这些气候？我在想

df_new = df[df['Climate'] in climate_of_interest]

但似乎不起作用。有什么建议吗？我不想把每一种气候都写出来作为它自己的条件，因为它很快就开始变得杂乱无章

Answer 1

这里有两种选择

isin

df[df.Climate.isin(climate_of_interest)]

  Date  Temperature Climate
0  4/1           50   Sunny
2  4/3           48   Rainy
4  4/5           33   Snowy

---

query

df.query('Climate in @climate_of_interest')

  Date  Temperature Climate
0  4/1           50   Sunny
2  4/3           48   Rainy
4  4/5           33   Snowy