如何在 for 循环中将结果写入多个 CSV 文件

Posted 2023-02-23

【中文标题】如何在 for 循环中将结果写入多个 CSV 文件

【英文标题】：how to write results in a for loop to multiple CSV files

【发布时间】：2016-08-29 03:59:07

【问题描述】：

我有

with open ('~/abc.csv', 'w') as f:
    write1 = csv.write(f)
    write1.writerow(['header1', 'header2', 'header3', 'header4'])

with open ('~/def.csv', 'w') as g:
    write2 = csv.write(g)
    write2.writerow(['header1', 'header2', 'header3', 'header4', 'header5', 'header6'])

for iteration in a_list:
    perform calculations
    result1 = ([h1, h2, h3, h4],[l1, l2, l3, l4],[m1, m2, m3, m4], ...,[])
    for pa in result1:
        write1.writerow(pa)

    def fun(result1):
        result2 = ([n1, n2, n3, n4, n5, n6],[p1, p2, p3, p4, p5, p6], [], ...[])
        for pb in result2:
             write2.writerow(pb)

期望两个 csv 文件为

'header1', 'header2', 'header3', 'header4'
h1, h2, h3, h4
l1, l2, l3, l4
m1, m2, m3, m4
      :

和

'header1', 'header2', 'header3', 'header4', 'header5', 'header6'
n1, n2, n3, n4, n5, n6
p1, p2, p3, p4, p5, p6

当所有迭代都完成时，这可以很容易地完成，并且可以使用 writer.writerows(pa) 轻松地将单独的 appended(list) result1 和 result2 写入单独的文件。但是，我想在每次迭代中写入 csv 文件，这样如果由于某种原因必须中断循环，我就不会错过。

谢谢！

【问题讨论】：

请edit您的问题并修正您的代码缩进。

【参考方案1】：

您可以在上下文管理器中放置两个文件并拥有两个 csv.writer 对象（从 Python 2.7 开始）：

with open ('~/abc.csv', 'w') as f, open ('~/def.csv', 'w') as g:
    writer1 = csv.writer(f)
    writer2 = csv.writer(g)

    writer1.writerow(['header1', 'header2', 'header3', 'header4'])    
    writer2.writerow(['header1', 'header2', 'header3', 'header4', 'header5', 'header6'])

    for iteration in a_list:
        # perform calculations
        for pa in result1:
            writer1.writerow(pa)
        for pb in result2:
            writer2.writerow(pb)

来自文档（对于 Python 2.7 到 3.5 都是相同的）。

如果有多个项目，上下文管理器的处理方式就像嵌套了多个 with 语句：

with A() as a, B() as b:
    suite

等价于

with A() as a:
    with B() as b:
        suite

这暗示了另一种方法，即让嵌套更深一层。

with open ('~/abc.csv', 'w') as f:
    with open ('~/def.csv', 'w') as g:
        writer1 = csv.writer(f)
        writer2 = csv.writer(g)
        ...

【讨论】：

with open ('~/abc.csv', 'w') as f, open ('~/def.csv', 'w') as g: 无效语法

您使用的是什么版本的 Python？这在 Python 2.7 中适用于我。

Python 版本 3.4.3

根据docs，它在3.4中也有效。

我错过了第一行。我也用于第二个文件。感谢您对杰米的支持！

【参考方案2】：

目前，您只声明了一个写入器变量。您需要第二个 writer 变量。

fWriter = csv.write(f)

...

gWriter = csv.write(g)

...

fWriter.writerow(pa)

...

gWriter.writerow(pb)

【讨论】：

我使用了不同的写入变量定义。但是，我在这里转换为示例代码时错过了它

您可以编辑示例代码，这样人们就不会将时间浪费在不是真正问题的事情上。

很抱歉。已编辑！