我有两个文本文件。第一个具有以下形式:
K, 6
J, 5
L, 4
第二个文件的格式为:
K_1, 6
K_2, 5
J_1, 4
J_2, 4
J_3, 5
L_1, 4
我需要以下形式的输出:
K_1, 6, 6, same
K_2, 5, 6, different
J_1, 4, 5, different
J_2, 4, 5, different
J_3, 5, 5, same
L_1, 4, 4, same
其中每行以第二个文本文件中的行开头(前两个值),然后我必须根据第一个文本文件中的子字符串来选择第三个值(例如,对于K_1,子字符串为K,我需要从第二个文本文件中选择值6第一个文本文件)。如果每行中的值相同,则应在每行上打印“相同”,否则应打印“不同”。
最后,我需要在输出文件中计算“相同”的行数和“不同”的行数。
我尝试编写以下代码,但未获得预期的输出:
m1 = open('TextFile_1.txt')
m2 = open('TextFile_2.txt')
result = open('Output.txt','w')
lookup_from = {}
l2=[]
for line1 in m1:
z1 = line1.split(',')[0].strip()
z2 = z1.split('_')[0].strip()
z3 = line1.split(',')[1].strip()
ZX = (z2, z1, z3)
lookup_from[ZX] = 0
for line2 in m2:
z11 = line2.split(',')[0].strip()
z22 = z11.split('_')[0].strip()
z33 = line2.split(',')[1].strip()
if z22 in [x for x,_,_ in lookup_from]:
z4 = (z22, z11, z33)
z5 = z4 + tuple([x for _,_,x in lookup_from])
l2.append(z5)
for i in l2:
result.write(str(i)[1:-1]+'\n')
result.close()
如果您创建一个简单的键,从第一个文件中获取值dict,然后在读取第二个文件时动态生成输出,则可以避免所有复杂的查找:
with open('TextFile_1.txt') as f1:
lookup = dict([x.strip() for x in line.split(',')] for line in f1)
with open('Output.txt', 'w') as out:
with open('TextFile_2.txt') as f2:
for line in f2:
k, v = [x.strip() for x in line.split(',')]
n = lookup[k.split('_')[0]]
out.write(', '.join(
[k, v, n, 'same' if v == n else 'different']) + '\n')
输出:
K_1, 6, 6, same
K_2, 5, 6, different
J_1, 4, 5, different
J_2, 4, 5, different
J_3, 5, 5, same
L_1, 4, 4, same
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句