1

我正在用 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()
4

2 回答 2

5

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

使用上下文管理器：

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退出块时文件将被关闭（即使是因为异常）。

于 2012-07-12T14:51:01.727 回答
1

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

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()
于 2012-07-12T15:08:56.643 回答