python - 用列表列表屏蔽两列(Pandas df)

Question

我有一个列表，其中包含恰好两个元素的列表。

我想将我的 DataFrame 过滤到列表列表中 [column_a, column_b] 的行。

示例:

输入:

l = [ [0,1], [1,2] ]
df = pd.DataFrame({'column_a' : [0, 1, 2], 'column_b' : [1, 3, 0]})

   column_a  column_b
0         0         1
1         1         3
2         2         0  

预期输出:

   column_a  column_b
0         0         1

Answer 1

从列表中创建一个 df 并合并:

df.merge(pd.DataFrame(l,columns=df.columns))

<小时/>

   column_a  column_b
0         0         1