在两列中用相同的值对数据帧进行子集设置

Question

我有一个数据帧，如下所示：

code_1	code_2
a1	a1
a2	a1
b1	b2
b3	b3

让我们尝试查询

代码

df.query('code_1 == code_2')

输出

code_1  code_2
0   a1      a1
3   b3      b3

如果我们希望索引是连续的

df.query('code_1 == code_2').reset_index(drop=True)

输出

code_1  code_2
0   a1      a1
1   b3      b3

有关查询的更多详细信息，请访问