pickle逐个写读

pickle 的妙用。逐个写入instance,逐个读取instance。
使用场景:逐个写入机器学习 训练的instance,训练时逐个读入buffer。

import pickle

# theory: pickle 如果是逐个 dump的那么久可以逐个读取
a=[i for i in range(9)]
# write
bw=open('tmp/a1.pt','wb')
for i in a:
    pickle.dump(i,bw)
bw.close()#必不可少
#read
br=open('tmp/a1.pt','rb')
try:
    print(pickle.load(br))#0
    print(pickle.load(br))#1
except EOFError:
    print('read out')

posted on 2021-11-19 11:02  likeatree  阅读(72)  评论(0)    收藏  举报

导航