在python中编写csv文件（文件在操作结束时留空）

Question

我正在用 python 写入 csv，我想知道为什么程序完成运行时我的输出文件是空白的。首先，这是我的代码：

reader = csv.reader(open(location, 'rU'), delimiter = ',', quotechar = '"')
writer = csv.writer(open('temp.csv', 'wb'), delimiter = ',', quotechar = '"')

writer.writerow(["test", "test", "test", "test", "test", "test", "test"])

count = 0
for row in reader:
    if count != 0 and count < len(split):
        writer.writerow([row[0], row[1], row[2], row[3], row[4], row[5], split[count-1]])
    count = count + 1

我知道一些事情：

• 程序完成后，写入“temp.csv”的所有内容都是“测试”行。文档的其余部分为空。
• 如果我在运行过程中强制停止程序，实际上会向文件写入文本：它只是由于某种原因而被擦除。
• 我做了一些研究，很多人发现“关闭” csv 文件可以解决他们的问题，但这似乎对我不起作用，除非我遗漏了一些明显的东西！

提前感谢您的帮助！

编辑：这是我正在使用的一些更新代码：

infile = open(location, 'rU')
outfile = open('temp.csv', 'wb')
reader = csv.reader(infile, delimiter = ',', quotechar = '"')
writer = csv.writer(outfile, delimiter = ',', quotechar = '"')
split = email_text.split('NEW MESSAGE BEGINS HERE')

count = 0
for row in reader:
    if count != 0 and count < len(split):
        writer.writerow([row[0], row[1], row[2], row[3], row[4], row[5], split[count-1]])
        outfile.flush()
    count = count + 1

infile.close()
outfile.close()

Answer 1

您没有关闭文件。虽然如果您只是在读取文件，这可能会很好，但在写入文件时这样做绝对不是一个好主意。

使用上下文管理器：

with open(location, 'rU') as infile, open('temp.csv', 'wb') as outfile:
    reader = csv.reader(infile, delimiter = ',', quotechar = '"')
    writer = csv.writer(outfile, delimiter = ',', quotechar = '"')
    ...

这可确保在退出 with 块时关闭文件（即使是因为异常）。

Answer 2

更改前两行以保存打开的文件句柄，然后包含关闭调用

reader_file = open(location, 'rU')
writer_file = open('temp.csv', 'wb')
reader = csv.reader(reader_file, delimiter = ',', quotechar = '"')
writer = csv.writer(writer_file, delimiter = ',', quotechar = '"')

# ... rest of code ...

reader_file.close()
writer_file.close()