如何编写合并其他 2 个 CSV 行的 CSV 文件 (Python)

Question

我正在尝试分析一些数据，为此，我通过写入一些由其他 CSV 文件组成的行来创建一个新的 CSV 文件。我已经从其中一个文件 (oldfile1) 中提取了数据，所以它是一个列表（带有我用来附加到新文件的特定索引），但另一个 (oldfile2) 我' m 用于文件的基础，因此我可以直接从该文件添加行，因为它们不需要过滤。新行的公式应为 row from oldfile2 + row from oldfile1。 first 是为了跳过注释行。然而，这段代码目前创建了一个非常大的输出文件 (200MB)——我怀疑它每行循环多次，复制写入的行。但是，我无法立即想到另一种方法来确保循环遍历 oldfile2 中的行，同时不复制写入的行。我也无法提供有关输出文件的更多详细信息，因为每当我尝试打开它时它都会崩溃。任何帮助表示赞赏。

with open('newfile.csv','w+') as f:
        reader = csv.reader(open('oldfile2.csv'), delimiter=',')
        writer = csv.writer(f, delimiter=',')
        first = next(reader)
        for oldrow2 in reader:
                outline = [oldrow2 + oldfile1[i] for i in oldfile1_indices]
                writer.writerow(outline)```

Answer 1

我无法测试，但我认为您需要 zip() 创建对 (oldrow2, i)，然后创建新行并保存

oldfile1 = list(csv.reader(open('oldfile1.csv'), delimiter=','))
oldfile1_indices = [...]

with open('newfile.csv','w+') as f:
    writer = csv.writer(f, delimiter=',')

    reader2 = csv.reader(open('oldfile2.csv'), delimiter=',')
    next(reader2)

    for oldrow2, i in zip(reader2, oldfile1_indices):
        outline = [oldrow2 + oldfile1[i]]
        writer.writerow(outline)

如何编写合并其他 2 个 CSV 行的 CSV 文件 (Python)

How to write a CSV file combining 2 other CSV rows (Python)

python

csv

file

file-handling