将数据框字典写入文件

C.Refsgaard

我有一本字典,字典中的每个键都有一个熊猫数据框。密钥之间的数据帧的长度不相等。

到达连接到每个键的数据帧需要花费一些时间,因此我希望将数据帧的字典保存到文件中,因此我可以将文件读入Python,而不必每次打开Python都运行脚本。

我的问题是:您如何建议将带有数据帧的字典写到文件中,然后再次读入?我尝试了以下操作,其中dictex是字典:

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

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

感谢您的时间。

关于独立保存数据帧而不使用SQL解决方案(或其他数据库格式)的规则,它可能是以下代码:

import csv
import pandas as pd 

def saver(dictex):
    for key, val in dictex.items():
        val.to_csv("data_{}.csv".format(str(key)))

    with open("keys.txt", "w") as f: #saving keys to file
        f.write(str(list(dictex.keys())))

def loader():
    """Reading data from keys"""
    with open("keys.txt", "r") as f:
        keys = eval(f.read())

    dictex = {}    
    for key in keys:
        dictex[key] = pd.read_csv("data_{}.csv".format(str(key)))

    return dictex

(...)

dictex = loader()

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章