- 製造業出身のデータサイエンティストがお送りする記事
- 今回はオブジェクトをpickleで保存/読み出しする方法をメモとして残しておきます。
##はじめに
オブジェクトを保存する際にpickleを使用して保存するのですが、その際にwith open
を使用して保存/読み出しを行っていたのですが、他の方法もある事を知りましたのでメモとして残しておきます。
1. with openで保存/読み出し
2. pandasで保存/読み出し
3. joblibで保存/読み出し
##with openで保存/読み出し
最初に私が良く使っていたwith open
を使用して保存/読み出しをする方法を整理します。
# データ作成
data = [a for a in range(0, 10)]
data
#[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
import pickle
#保存
with open("data.pkl", "wb") as f:
pickle.dump(data, f)
#読み出し
with open("data.pkl", "rb") as f:
hoge = pickle.load(f)
hoge
#[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
##pandasで保存/読み出し
次はpandas
を使用して保存/読み出しをする方法を整理します。
import pandas as pd
# 保存
pd.to_pickle(data, "data_1.pkl")
#読み出し
hoge = pd.read_pickle("data_1.pkl")
hoge
#[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
##joblibで保存/読み出し
最後にjoblib
を使用して保存/読み出しをする方法を整理します。
import joblib
# 保存
joblib.dump(data, "data_2.jb", compress=3)
#読み出し
hoge = joblib.load("data_2.jb")
hoge
#[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
##さいごに
最後まで読んで頂き、ありがとうございました。
簡単ですが、オブジェクトをpickleで保存/読み出しする方法をメモとして残しておきました。
訂正要望がありましたら、ご連絡頂けますと幸いです。