首页 > 解决方案 > Python将两个不同文件中的行合并到一个列表中

问题描述

我正在尝试从两个不同文件的行中构建一个列表。

文件1.txt

['2.02', '-3.88', '15.25']
['4.40', '-2.36', '14.97']
['5.44', '0.34', '7.34']
['5.76', '0.41', '7.60']
['5.35', '0.19', '13.95']

文件2.txt

['P1']
['P2']
['P3']
['P4']
['P5']

理想的输出

['2.02', '-3.88', '15.25', 'P1']
['4.40', '-2.36', '14.97', 'P2']
['5.44', '0.34', '7.34', 'P3']
['5.76', '0.41', '7.60', 'P4']
['5.35', '0.19', '13.95', 'P5']

我试图简单地添加行

combined = open('file1.txt').readlines() + open('line2.txt').readlines()

输出

["['2.02', '-3.88', '15.25']\n", "['4.40', '-2.36', '14.97']\n", "['5.44', '0.34', '7.34']\n", "['5.76', '0.41', '7.60']\n", "['5.35', '0.19', '13.95']\n", 'P1\n', 'P2\n', 'P3\n', 'P4\n', 'P5\n']

我还编写了一个函数,试图将两者合并到第三个文件 file3.txt

def combiner():
with open("file1.txt", 'r') as file1:
    with open('file2.txt', 'r') as file2:
        with open('file3.txt',"w") as file3:
            clines = file1.readlines()
            plines = file2.readlines()
   #Write to third file
            for i in range(len(clines)):
                line = clines[i].strip() + plines[i]
                file3.write(line)

但它仍然写在 2 个不同的列表中

猫文件3.txt

['2.02', '-3.88', '15.25']['P1']
['4.40', '-2.36', '14.97']['P2']
['5.44', '0.34', '7.34']['P3']
['5.76', '0.41', '7.60']['P4']
['5.35', '0.19', '13.95']['P5']

最好的方法是返回这样一个列表的函数。

标签: pythonlistfilemerge

解决方案


尽管文件中的行类似于列表,但您实际上将它们视为字符串。您可以使用ast.literal_eval将字符串评估为列表,组合它们,然后将它们写入另一个文件

from ast import literal_eval

def combiner(infile1, infile2, outfile):
    with open(infile1) as f1, open(infile2) as f2, open(outfile, 'w+') as f3:
        f1_lists = (literal_eval(line) for line in f1)
        f2_lists = (literal_eval(line) for line in f2)
        for l1, l2 in zip(f1_lists, f2_lists):
            f3.write(str(l1+l2) + '\n')  # This could also be print(l1+l2, file=f3)

推荐阅读