问题描述
文件1
71528
1452
14587
文件2
country_hoj_17458 9 CA 5 CA 78.4
country_hoj_1452 10 CA 15 CA 96.5
country_hoj_14787 19 CA 51 CA 12.4
country_hoj_15742 19 CA 51 CA 12.4
country_hoj_171528 19 CA 51 CA 12.4
我想在第一列中打印出模式(编号)文件1与文件2匹配的行。我想要这样的文件
country_hoj_1452 10 CA 15 CA 96.5
country_hoj_14787 19 CA 51 CA 12.4
我的脚本是这样的:
filename = numbers.txt
filename2 = data.txt
with open(filename) as f:
with open (filename2) as m:
for line in f:
if line in m:
print (m)
解决方法
filename = 'numbers.txt'
filename2 = 'data.txt'
with open(filename) as numberLines:
with open (filename2) as dataLines:
nL = numberLines.read().splitlines()
dL = dataLines.read().splitlines()
dataReadLines = [j for i in nL for j in dL if i in j]
#dataReadLines = [i for i in nL]
print (str(dataReadLines))
另一个答案,其中每个键都与各自的数据配对。我更改了您的输入,使用以下代码即可轻松理解。
from collections import defaultdict
filename = 'numbers.txt'
filename2 = 'data.txt'
with open(filename) as numberLines:
with open (filename2) as dataLines:
nL = numberLines.read().splitlines()
dL = dataLines.read().splitlines()
defDList = defaultdict(list)
dataReadLines = [defDList[i].append(j) for i in nL for j in dL if i in j]
#dataReadLines = [i for i in nL]
print (defDList)