サブロウ丸

Sabrou-mal サブロウ丸

主にプログラミングと数学

Python pickle プロトコルの指定

今までpickleファイルを作成するときは、

with open(file, 'wb') as f:
    pickle.dump(obj=obj, file=f)

としていましたが、プロトコルバージョンを高いものに指定してやると、より速く読み込めるpickleファイルが作れるらしい。。

最新のプロトコルバージョンは、pickle.HIGHEST_PROTOCOLで確認できます。

with open(file, 'wb') as f:
    pickle.dump(obj=obj, file=f, protocol=pickle.HIGEST_PROTOCOL)


もしくは、protocolに負の整数を指定してやります。(その場合もprotocolにはHIGEST_PROTOCOLが採用されます)

with open(file, 'wb') as f:
    pickle.dump(obj=obj, file=f, protocol=-1)



ただ、バージョンを高くすると、python2系との互換性がなくなる場合があるので、注意してください。