python - 如何在 pandas 数据框中使用字符串列表作为条件

Question

这个问题已经有答案了:

How to check if a value is in the list in selection from pandas data frame? [duplicate] (2 个回答)

已关闭 4 年前。

如果我有一个如下所示的数据框(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

<小时/>

查询

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