python - 如何在Python中的数据集的两个不同列中找到相同值的匹配

Question

我有一个 .txt 文件格式的数据集，如下所示:

# ID M R x y z
  0 1 200 2 2 2 
  1 5 300 2 4 6 
  2 5 600 2 4 2
  3 4 500 2 2 5
  4 4 500 3 2 1
  5 7 300 2 4 1

我想打印 x 列和 y 列中具有相同值的行的 ID

我写了一篇文章，但它没有达到我想要的效果:

import numpy as np
halo = 'test.txt'
ID, m,r,x,y,z= np.loadtxt(halo)


for yv in np.unique(halo[3] and halo[4]):
    if yv != np.nan:
        idx = x == yv

print idx

基本上我想要如下的输出:

[0, 3]
[1, 2, 5]

其中包含 x 和 y 中具有相同值的 ID 列值。非常感谢您的帮助。

Answer 1

这可以用 pandas 来完成，

import pandas as pd
df = pd.read_csv('path_to_your_txt_file')
rows = df[df['x'] == df['y']]

其中 rows.index = [0, 3]