python - 将数据帧字典写入文件

Question

我有一本字典，对于我字典中的每个键，我都有一个 pandas 数据框。从键到键的数据帧长度不等。

获取连接到每个键的数据帧需要一些时间，因此我希望将我的数据帧字典保存到一个文件中，这样我就可以将文件读入 Python 而不是每次运行我的脚本打开 Python。

我的问题是:您建议如何将带有数据框的字典写入文件 - 并再次读入？我尝试了以下方法，其中 dictex 是字典:

w = csv.writer(open("output.csv", "w"))
for key, val in dictex.items():
    w.writerow([key, val])

但我不确定我是否得到了我想要的，因为我很难再次将文件读入 Python。

感谢您的宝贵时间。

Answer 1

您可以使用 pickle 在 python 中保存数据帧字典。

import pickle

df1 = pd.DataFrame(data={'a':[1,2,3], 'b':[4,5,6]})
df2 = pd.DataFrame(data={'a':[5,5,5,5,5], 'b':[5,5,5,5,5]})

d = {}
d['df1'] = df1
d['df2'] = df2

with open('dict_of_dfs.pickle', 'wb') as f:
    pickle.dump(d, f)