代码之家 › 专栏 › 技术社区 › PabloG

如何在多列上同时按条件列表筛选数据帧行

multiple-columns filter pandas python

PabloG · 技术社区 · 3 年前

我有一个DataFrame,我想在其中过滤同时适用于两列的条件列表的行。例如:

pd.DataFrame({'A':['Y','Y','Y','Z','Z'],'B':[1,2,2,3,4],'C':[100,200,234,358,4243]})
   A  B     C
0  Y  1   100
1  Y  2   200
2  Y  2   234
3  Z  3   358
4  Z  4  4243
filter_on=[('Y',2),('Z',4)]

过滤时在哪里 A 和 B 有两个值中的任何一个 过滤器打开 ,我应该得到以下数据帧:

   A  B     C
1  Y  2   200
2  Y  2   234
4  Z  4  4243

3 回复 | 直到 3 年前

BENY 3 年前

尝试 isin 转换为后 tuple

out = df[df[['A','B']].apply(tuple,1).isin(filter_on)].copy()
   A  B     C
1  Y  2   200
2  Y  2   234
4  Z  4  4243

Dani Mesejo 3 年前

你可以建立一个 MultiIndex 从列和使用 isin 计算遮罩,如下所示:

mask = pd.MultiIndex.from_arrays([df['A'], df['B']]).isin(filter_on)
print(df[mask])

输出

   A  B     C
1  Y  2   200
2  Y  2   234
4  Z  4  4243

Caner Burc BASKAYA 3 年前

data.iloc[[i for  i,(a,b) in enumerate(zip(data.A, data.B)) if (a,b) in filter_on]]

cs95 3 年前

这个简洁的怎么样 set_index 和 Index.isin 基于Web的解决方案?

df[df.set_index(['A', 'B']).index.isin(filter_on)]

   A  B     C
1  Y  2   200
2  Y  2   234
4  Z  4  4243

推荐文章

Mainland · Python数据帧规范化值错误:列的长度必须与键相同

1 年前

user026 · 如何根据特定窗口的平均值(行数)创建新列?

1 年前

rpn · 如何在列[1]中连续第二次出现“0”时返回列[0]的值

1 年前

asmgx · 为什么合并数据帧不能按照python中的预期方式工作

1 年前

Gtoth · 如何分割Pandas DataFrame中包含多个日期的两个时间戳之间的差异

1 年前

Domarius · 使用loc为多行设置多列值

1 年前

Swastik Bhattacharyya · 如何在同一类别类型的多列上运行get_dummies()函数?

1 年前

DrZoidberg09 · 如何在字典列表中创建一个新关键字,该关键字是另一个关键字的总和?

1 年前

armstrong3701 · 如何有效地处理熊猫数据框中缺失的数据并计算条件统计?

1 年前

msts1906 · 大熊猫向乳胶的适当多品种出口

1 年前