python将dictreader写入csv文件

Question

我正在使用python 2.4并尝试将unix last命令的值导出到csv文件。 我无法弄清楚如何将其实际写入每一行到csv文件，任何帮助将不胜感激！

import csv

def check(user,logfile,name):
    logfile.write('********' + name + '*********\n')
    g = subprocess.Popen(["last",user], stdout=subprocess.PIPE)
    stdout, stderr = g.communicate()
    reader = csv.DictReader(stdout.splitlines(),
                            delimiter=' ', skipinitialspace=True,
                            fieldnames=['id', 'pts', 'cpu',
                                        'day', 'month', 'date',
                                        'time', 'dash', 'off',
                                        'loggedin', 'test1', 'test2'])

    writer = csv.writer(open('dict.csv','wb'))
    for row in reader:
        writer.writerow(row)

Answer 1

您需要使用csv.DictWriter()代替（具有匹配的字段名），或将字典行变成一个序列：

writer.row([value for key, value in sorted(row.items())])

例如，将输出按其键排序的值。

使用DictWriter可能很简单：

writer = csv.DictWriter(open('dict.csv','wb'), fieldnames=reader.fieldnames)

它将以与DictReader()类所期望的顺序相同的顺序写入完全相同的字段。