Skip to content

[Python] 使用 Pickle 模組保存資料(持久化數據)

Last Updated on 2021-06-19 by Clay

今天我想要紀錄的是 pickle 這個模組。首先要說明的是,pickle 在 Python 中是相當相當有名的,它的運作方法跟 Json 有點像,都是可以壓縮資料、保存、並且隨時隨地解開我們儲存好的資料,還原我們 Python 的工作階段。

不過由於它是專門為 Python 而生的、而非像 Json 那般是『輕量級的資料儲存格式』,故若是需要跨語言進行開發時,pickle 這種儲存資料的格式可能就不太適合,不過若只是要進行 Python 專案的開發,那麼 pickle 可是非常非常方便的!

以下來簡單看個實際使用 pickle 的例子。


使用 pickle 儲存、讀取資料

由於 pickle 本身便是 Python 當中內建的模組,所以我們不需要使用額外的指令來安裝它,直接在程式中匯入即可。

我們來看段簡單的資料儲存:

import pickle

my_dict = {
    'a': 1,
    'b': 2,
    'c': 3,
}

with open('test.pickle', 'wb') as f:
    pickle.dump(my_dict, f)



我們可以使用 "dump()" 函式來儲存我們的資料,以上方程式碼為例,便是我的 "my_dict" 的資料。

以下再來紀錄使用 "load()" 讀取儲存好的 pickle 資料。

# Load
with open('test.pickle', 'rb') as f:
    new_dict = pickle.load(f)

print(new_dict)



Output:

{'a': 1, 'b': 2, 'c': 3}

這樣就簡單地將資料讀取進來了!

當然,熟悉 Json 的人可能會想:這樣跟 Json 有什麼不一樣?不都是儲存資料讀取資料嗎?連 Function 都一模一樣欸!

但不是這樣的,pickle 由於是專門支援 Python 的,故就像我上方所提及的一般,可以還原我們的『Python 工作階段』。

class dog:
    def __init__(self, color):
        self.color = color

puppy = dog('black')


with open('test.pickle', 'wb') as f:
    pickle.dump(puppy, f)



假設我們今天定義了一個 Class: dog,並建立 "color" 這個屬性。接著我們將 "puppy" 這個變數物件化,並賦予 "black" 這個顏色的屬性。

然後我們使用 pickle 儲存起來,這就是 Json 難以做到的了,Json 僅能支援部份的 Python 資料型態。

# Load
with open('test.pickle', 'rb') as f:
    my_dog = pickle.load(f)

print(my_dog.color)



Output:

black

我們可以看到,pickle 完美地還原了我們自定義的 Class,印出了 my_dog 的顏色,是不是很方便呢?

不過聽說 Json 的讀取速度還是比 pickle 快的。如果是進行大量資料的儲存讀取,或許還是使用 Json 會更好的一點。我也希望之後有空的時間能拿比較大量的資料自己做做看實驗。


References


Read More

Tags:

Leave a Reply取消回覆

Exit mobile version