在python中的csv行中查找相同的值

Question

我有一个代码，可以在csv文件中查找在同一行中彼此小数点后1.0位以内的数字。不过，当我运行它的时候，它会打印所有的东西。不仅仅是具有我想要的条件的行，即第2列和第3列中的值彼此在1.0范围内。我想运行代码并让它显示，第一列（记录的时间或者更好的是列号），第二列和第三列，因为它们应该在1.0之内。数据文件如下所示：

Time    Chan1       Chan2
04:07.0 52.31515503 16.49450684
04:07.1 23.55230713 62.48802185
04:08.0 46.06217957 24.94955444
04:08.0 41.72077942 31.32516479
04:08.0 19.80723572 25.73182678

这是我的密码：

import numpy as np
from matplotlib import *
from pylab import *

filename = raw_input("Enter file name: ") + '.csv'
filepath = '/home/david/Desktop/' + filename


data = np.genfromtxt(filepath, delimiter=',', dtype=float)



first=[row[0] for row in data]
rownum1=[row[1] for row in data]
rownum2=[row[2] for row in data]



for row in data:
    if ((abs(row[1]-row[2]) <= 1.0)):
        print("The values in row 0 are 1 and 2, are within 1.0 of each other.", first, rownum1, rownum2)

这是我的输出：

26.3460998535, 44.587371826199998, 42.610519409200002, 24.7272491455, 89.397918701199998, 25.479614257800002, 30.991180419900001, 25.676086425800001

但我想把它作为一个输出：

4:09.0, 23.456, 22.5

Answer 1

你可以这样做：

data = np.genfromtxt(filepath, names=True, dtype=None)
idx = np.abs(data['Chan1'] - data['Chan2'])<1
print data[idx]