python - 使用键作为列标题将 python 字典写入 CSV 文件

Question

我正在尝试将字典中的元素写入一个文本文件，其中每个键都是一列。目前我有一些看起来像的东西

import csv
import numpy as np



data1 = np.arange(10)
data2 = np.arange(10)*2
data3 = np.arange(10)*3

writefile = '../Desktop/data.txt'
datadict = {} 

datadict['data1'] = data1
datadict['data2'] = data2
datadict['data3'] = data3


f = open( writefile, 'w' )
fieldnames = ['data1','data2', 'data3']
data = csv.DictWriter(writefile, fieldnames, restval='', extrasaction='ignore', dialect='excel')

f.close()

但它给了我错误“argument 1 must have a "write"method”。我不确定那是什么意思。我也担心 dialect = 'excel'，但我不确定还要放什么。最后我想要一个看起来像这样的文件:

谢谢

Answer 1

这里根本不需要使用 DictWriter:

import csv
import numpy as np

data1 = np.arange(10)
data2 = np.arange(10)*2
data3 = np.arange(10)*3

writefile = '../test.csv'
fieldnames = ['data1','data2', 'data3']
with open( writefile, 'w' ) as f:
    writer = csv.writer(f)
    writer.writerow(fieldnames)
    writer.writerows(zip(data1, data2, data3))