Python Pickle 在同一个文件中序列化多个对象

最新推荐文章于 2024-10-17 16:14:23 发布

青盏

最新推荐文章于 2024-10-17 16:14:23 发布

阅读量5.5k

点赞数 3

分类专栏： python

本文链接：https://blog.csdn.net/qq_16234613/article/details/82628606

版权

python 专栏收录该内容

22 篇文章 5 订阅

订阅专栏

也是看别人代码才知道可以打开一个文件就可以把多个对象序列化到这个文件中。

with open('../raw_data/remap.pkl', 'wb') as f:
  pickle.dump(reviews_df, f, pickle.HIGHEST_PROTOCOL) # uid, iid
  pickle.dump(cate_list, f, pickle.HIGHEST_PROTOCOL) # cid of iid line
  pickle.dump((user_count, item_count, cate_count, example_count),
              f, pickle.HIGHEST_PROTOCOL)
  pickle.dump((asin_key, cate_key, revi_key), f, pickle.HIGHEST_PROTOCOL)

with open('../raw_data/remap.pkl', 'rb') as f:
  reviews_df = pickle.load(f)
  cate_list = pickle.load(f)
  user_count, item_count, cate_count, example_count = pickle.load(f)