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

Question

I have

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

这可以很容易地完成，当所有迭代都完成后，可以将单独的附加（列表）result1和result2轻松地写入到不同的文件中，使用writer.writerows(pa)。然而，我想在每次迭代时写入csv文件，这样如果必须由于某种原因中断循环，我就不会错过任何内容。

谢谢！

- Jagruth

---

请编辑您的问题并修复代码的缩进。 - Chris

Answer 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

is equivalent to

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)
        ...

- Jamie Bull

---

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

你使用的是哪个版本的Python？这在Python 2.7中对我有效。 - Jamie Bull

Python版本3.4.3 - Jagruth

根据文档，在3.4版本中也是有效的。 - Jamie Bull

我错过了第一行。我也在第二个文件中使用了with。谢谢你的支持，Jamie！ - Jagruth

可能是我的问题。那是我最初放置的内容，没有注意到。很高兴它对你有用。 - Jamie Bull

Answer 2

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

fWriter = csv.write(f)

...

gWriter = csv.write(g)

...

fWriter.writerow(pa)

...

gWriter.writerow(pb)

- Chris G

---

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

您可以编辑您的示例代码，以便人们不会浪费时间在不是真正问题的事情上。 - Jamie Bull

抱歉，已经编辑好了！ - Jagruth