Python最簡單的永久存儲：pickle模組

• 2020 年 5 月 11 日
• AI
• Python

現在深度學習很多使用的是動態圖，方便調試程式

但是隨之也帶來了一個問題，就是如果程式中途，我們關閉了運行環境

那再進入以後，中間變數就沒了，要從頭運行一遍

對於深度學習來說，很多數據的處理也能是批量進行，中間變數如果一旦損失，那非常浪費時間

那最好的方法就是把記憶體中的數據暫時存放到硬碟上，這樣就斷電也不丟失啦！

在 Python 中，雖然有很多存儲模組，但是我覺得最簡單的就是 pickle 模組了

程式案例：

import pickle
import numpy as np

a = np.array([[1,2,3],[4,5,6]])
b = 'www'
c = [a,b]

f = open('data.pk','wb')
pickle.dump(c, f)
f.close()

fr = open('data.pk','rb')
d = pickle.load(fr)
fr.close()
print(d)

運行結果：

[array([[1, 2, 3],
       [4, 5, 6]]), 'www']