python pickleデータを読み書きする

pickle化されたファイルを使用する機会があったので備忘録。

目次

• testデータを作成（データを保存する）
• pickleでpickleデータをを読み込む
• pandasでpickleデータを読み込む
• コマンドラインでpickleデータを確認する

testデータを作成（データを保存する）

• pickleはpythonの標準モジュール
• pickleをimportする
• 保存ファイルを開いてpickle.dump()で保存

#! /usr/bin/python3

import pickle

# データを作成
data = ["data{}".format(i) for i in range(10)]
print(data)

# データを保存
with open('test.pkl', 'wb') as f:
    pickle.dump(data, f)

pickleでpickleデータをを読み込む

• ファイルを開いてpickle.load()で読み込み

#! /usr/bin/python3

import pickle

# データを読み込み
with open('test.pkl', 'rb') as f:
    load_data = pickle.load(f)

print(load_data)

pandasでpickleデータを読み込む

• pandas はpipでインストール
• pandasをimportする
• pandas.read_pickleで読みこむ

import pandas as pd

df_pkl = pd.read_pickle('test.pkl')

print(df_pkl)

コマンドラインでpickleデータを確認する

• test.pklにファイルパスを指定すればコマンドライン上で確認できる。
• 中身を見るだけならこれでOK

python3 -m pickle test.pkl

• よく使う場合はaliasに登録すると楽。

# ~/.bashrc

alias pcat='python3 -m pickle'

以上。