python - 如何从 Python 字典创建自定义 CSV 文件？

Question

我想像这样创建一个 CSV 文件:

Features, Value
f1, 1
f2, 2
f3, 3

但是我明白了:

f1, f2, f3
1, 2, 3

我写的代码:

my_dict = {'f1': 1, 'f2': 2, 'f3': 3}
with open('testcsv.csv', 'w') as f:
    w = csv.DictWriter(f, my_dict.keys())
    w.writeheader()
    w.writerow(my_dict)

我该怎么办？

Answer 1

您需要先根据预期的结构转换您的数据

data = [{'Features':i,'Value':my_dict[i]} for i in my_dict]

一旦你这样做了，你就可以像这样使用 pandas 将它保存为 CSV:

import pandas as pd
df = pd.DataFrame(data)
df.to_csv('testcsv.csv',index=None)

生成的 csv:

Features,Value
f1,1
f2,2
f3,3