找到匹配线后的第3行

时间:2016-04-26 19:55:01

标签: python-2.7 pandas text-files

我有用于搜索的txt文件和用于引用的csv文件。我需要从txt文件中的csv文件中搜索每个项目。但是,我只想保存匹配行的第三行。

txt文件的意思是:

SCJ 149  NAME ....
 PROGRAM_FILE...
 WTT_FILE...
SCJ 146  NAME...
 PROGRAM_FILE...
 WTT_FILE...
.
.
.
.
SCJ 1522 NAME ....
 PROGRAM_FILE...
 WTT_FILE...

引用文件是:

SCJ 1
SCJ 2
SCJ 3
SCJ 4
SCJ 5
.
.
.
.
SCJ 2042

所以我想从引用文件中选择每个项目,并匹配txt文件中的SCJ XXX代码,但我想在匹配代码后得到第三行信息。

我将代码编程如下:

import pandas as pd
aaa = pd.read_csv('refer.csv')
aaa = aaa.as_matrix()
for j in range(len(aaa)):
    with open ('data.txt','r') as root:    
        for i in root:
            if str(aaa[j][0]) + '  NAME' in i:
                lines = [root.readline() for i in range(3)]

但显示错误Mixing iteration and read methods would lose data

任何人都可以帮我这个吗?

0 个答案:

没有答案