python - 在 python 中保存列表

Question

我正在尝试将输出导出到列表。这是我的代码:

import csv
import numpy as np

row_list = np.arange(0,5)
for row in row_list:
    a = row + 3
    b = row + 4
    c = row + 5


    result = [a, b, c]
    csvfile = "/home/Desktop/test"

    with open(csvfile, "w") as output:
        writer = csv.writer(output, lineterminator='\t')
        for val in result:
            writer.writerow([val])

我运行代码，在我的桌面上创建了一个测试文件(这正是我想要的)，但文件中的数据是错误的。输出是: 7 8 9

但这不是我想要的。该脚本正在循环运行，但仅导出最后一行。关于如何让它通过以下方式导出文件中的数据的任何想法:

3 4 5
4 5 6
5 6 7
6 7 8
7 8 9

谢谢!

编辑:如果我想把这个脚本交给我的教授怎么办？她的笔记本电脑上不存在目录 /home/paula/Desktop/test。我尝试编写 csvfile = "~/Desktop/test" 但它给了我一个错误。任何想法？

注意:我使用的是 ubuntu 12.04。

Answer 1

当您以 w 模式打开文件时，它会删除所有现有内容。即使没有，您也不会在行尾写入换行符。

打开文件一次，并使用更明智的“行”概念 - 对应于数据的实际行:

with open(csvfile, "w") as output:
    writer = csv.writer(output, delimiter='\t')

    for row_num in row_list:
        a = row_num + 3
        b = row_num + 4
        c = row_num + 5

        row = [a, b, c]
        writer.writerow(row)