将字典写入 csv 文件以供再次使用

Question

我有一本字典：

inputdict = {key1: [value1,value2..], key2: [value1,value2..], key3: [value1,value2..]}

我想将数据写入 dict.csv 文件，格式如下：

key1: [value11,value12,..]
key2: [value1,value22,..]
key3: [value31,value32,..]

我写道：

import csv
with open(ouputfile, 'wb') as csv_file:
        writer = csv.writer(csv_file)
        for key, value in inputdict.items():
           writer.writerow([key: value])

但现在我有一个错误：

 TypeError: a bytes-like object is required, not 'str'

当我设法写出这样的文件时，我也想将它读回新字典。

字典包含键和整数列表中的字符串。我想通过读取 csv 文件来重用其他脚本的字典。

Answer 1

使用w 而不是wb 写入文件。并且在回读时使用r 而不是rb。

那么你可以这样做：

with open('/path/to/input_dict.csv','w') as fou:
    writer = csv.writer(fou)
    for k,v in inputdict.iteritems():
        writer.writerow(["{'%s':%s}" % (k,v)])

然后再读一遍：

import ast
with open('/path/to/input_dict.csv','r') as fin:
    for line in fin:
        ast.literal_eval(ast.literal_eval(line))

但是，您应该能够使用pickle 更轻松地做您想做的事情：

import pickle
with open('/path/to/input.csv', 'wb') as fou:
    pickle.dump(inputdict,fou)

阅读：

with open('/path/to/input.csv', 'rb') as fin:
    inputdict = pickle.load(fin)