如何在熊猫数据框中配置过滤器?

时间:2021-01-05 04:43:11

标签: python pandas dataframe filter

我有一个 df:

  col1 col2
0  A   23
1  B   25
2  C   26
3  D   92

现在我需要在 col1 上过滤这个 df 以获得 A 和 C 的值。所以我执行以下操作:

filter1 = (df['col1'] == 'A') | (df['col1'] == 'C')
new_df = df[filter1]

这将为我提供第 0 行和第 2 行的 df。但是,我如何使其可配置?例如,目前我们正在对 A 或 C 进行过滤,也许将来我需要包含 D,或包含 B 等。我该如何编写它以便我的过滤器可配置,并且只需更改某些内容和 df 列表中的值过滤器是否更新?

1 个答案:

答案 0 :(得分:1)

yourList=["All the values you need to be filtered"]


df[df['col1'].isin(yourList)]
相关问题