用两列中相同的值对数据框进行子集化

Question

我有一个如下所示的数据框：

代码_1代码_2a1a1a2a1b1b2b3b3

我在这里想做的是，我想通过选择“code_1”和“code_2”中具有相同值的行来对数据帧进行子集化

最终输出如下所示：

代码_1代码_2a1a1b3b3

谢谢你

Answer 1

让我们尝试一下

查询

代码

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

更多查询详情请参阅

这里