快捷導(dǎo)航

關(guān)于如何把Python對象存儲為文件的方法詳解

更新時間：2024年01月25日 11:32:26 作者：微小冷

本文將給大家介紹如何把Python對象存儲為文件的方法,pickle可以用二進(jìn)制表示并讀寫python數(shù)據(jù),這個功能并不安全,如果把一個pickle暴露給別人,有被植入惡意程序的風(fēng)險,文中通過代碼給大家講解的非常詳細(xì),需要的朋友可以參考下

加載和保存

pickle中有兩組加載和保存函數(shù)

	保存	加載
交互文件	dump	load
交互字節(jié)碼	dumps	loads

例如

x = [i**2 for i in range(10)]
import pickle
s = pickle.dumps(x)
print(s)
# b'\x80\x04\x95\x19\x00\x00\x00\x00\x00\x00\x00]\x94(K\x00K\x01K\x04K\tK\x10K\x19K$K1K@KQe.'
with open("test.pickle", 'wb') as f:
   pickle.dump(x, f)        # 將x存為f

將x存儲為文件之后，還可以重新打開

with open("test.pickle", 'rb') as f:
    y = pickle.load(f)

print(y)
# [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

注意事項

除了不夠安全之外，使用pickle另外一個需要注意的是，到目前為止，pickle已經(jīng)有6個版本，其中v0和v1過于古老，基本沒什么能用的地方；v3以后的版本則不再支持Python2，下表列出了自v2到v5的一些更新特性

	引入版本	文檔
v2	Python 2.3	PEP307	為新式類提供了更高效的封存機(jī)制。
v3	Python 3.0		顯式地支持字節(jié)對象為Python 3.0-3.7 的默認(rèn)協(xié)議。
v4	Python 3.4	PEP 3154	支持存儲較大對象，并針對數(shù)據(jù)格式進(jìn)行了優(yōu)化 Python 3.8使用的默認(rèn)協(xié)議。
v5	Python 3.8	PEP 574	增加對帶外數(shù)據(jù)的支持，并可加速帶內(nèi)數(shù)據(jù)處理

理論上講，pickle可以封裝一切python對象，所以對于一些比較耗時的算法，可以通過pickle來保存當(dāng)前的狀態(tài)。但要注意，盡管pickle可以封存函數(shù)，但并不能封存匿名函數(shù)

test = lambda x : x
s = pickle.dumps(test)  # 報錯
def test(x):
    print(x)

s = pickle.dumps(test)
print(s)
#b'\x80\x04\x95\x15\x00\x00\x00\x00\x00\x00\x00\x8c\x08__main__\x94\x8c\x04test\x94\x93\x94.'

copyreg

python標(biāo)準(zhǔn)庫中還提供了copyreg模塊，可以注冊一個封存函數(shù)，在對某個對象進(jìn)行封存時可調(diào)用。示例如下

import copyreg, pickle
class C:
    def __init__(self, a):
        self.a = a

def pickle_c(c):
    print("您封存了一個C對象")
    return C, (c.a,)

copyreg.pickle(C, pickle_c)
c = C(1)

p = pickle.dumps(c)
# 您封存了一個C對象

在通過copyreg.pickle對類型C注冊函數(shù)pickle_c之后，再去用pickle.dumps對c的對象進(jìn)行封存，就會調(diào)用pickle_c，從而打印出預(yù)設(shè)的字符串。

在copyreg.pickle中注冊的函數(shù)，必須返回一個字符串或包含二至六個元素的元組。

命令行工具

python標(biāo)準(zhǔn)庫還提供了pickle的命令行工具，調(diào)用格式如下，其中x.pickle是一個封裝了python對象的二進(jìn)制文件。

python -m pickle x.pickle

常見的命令行選項如下表

	指令	說明
-a	–annotate	使用簡短的操作碼描述來標(biāo)注每一行。
-o	–output=<file>	輸出應(yīng)當(dāng)寫入到的文件名稱。
-l	–indentlevel=<num>	一個新的 MARK 層級所需縮進(jìn)的空格數(shù)。
-m	–memo	當(dāng)反匯編多個對象時，保留各個反匯編的備忘記錄。
-p	–preamble=<preamble>	當(dāng)指定一個以上的 pickle 文件時，在每次反匯編之前打印給定的前言。