快捷導(dǎo)航

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

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

本文將給大家介紹如何把Python對(duì)象存儲(chǔ)為文件的方法,pickle可以用二進(jìn)制表示并讀寫python數(shù)據(jù),這個(gè)功能并不安全,如果把一個(gè)pickle暴露給別人,有被植入惡意程序的風(fēng)險(xiǎn),文中通過(guò)代碼給大家講解的非常詳細(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存儲(chǔ)為文件之后，還可以重新打開

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

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

注意事項(xiàng)

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

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

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

test = lambda x : x
s = pickle.dumps(test)  # 報(bào)錯(cuò)
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)庫(kù)中還提供了copyreg模塊，可以注冊(cè)一個(gè)封存函數(shù)，在對(duì)某個(gè)對(duì)象進(jìn)行封存時(shí)可調(diào)用。示例如下

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

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

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

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

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

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

命令行工具

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

python -m pickle x.pickle

常見的命令行選項(xiàng)如下表

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