首页 > 解决方案 > 如何在熊猫数据框中使过滤器可配置?

问题描述

我有一个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 列表中的值过滤器更新了吗?

标签: pythonpandasdataframefilter

解决方案


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


df[df['col1'].isin(yourList)]

推荐阅读