È possibile utilizzare il picklemodulo nella libreria standard. Ecco una sua elementare applicazione al tuo esempio:
import pickle
class Company(object):
def __init__(self, name, value):
self.name = name
self.value = value
with open('company_data.pkl', 'wb') as output:
company1 = Company('banana', 40)
pickle.dump(company1, output, pickle.HIGHEST_PROTOCOL)
company2 = Company('spam', 42)
pickle.dump(company2, output, pickle.HIGHEST_PROTOCOL)
del company1
del company2
with open('company_data.pkl', 'rb') as input:
company1 = pickle.load(input)
print(company1.name) # -> banana
print(company1.value) # -> 40
company2 = pickle.load(input)
print(company2.name) # -> spam
print(company2.value) # -> 42
Puoi anche definire la tua semplice utility come la seguente che apre un file e scrive un singolo oggetto su di esso:
def save_object(obj, filename):
with open(filename, 'wb') as output: # Overwrites any existing file.
pickle.dump(obj, output, pickle.HIGHEST_PROTOCOL)
# sample usage
save_object(company1, 'company1.pkl')
Aggiornare
Dal momento che questa è una risposta così popolare, vorrei toccare alcuni argomenti di utilizzo leggermente avanzati.
cPickle(o _pickle) vspickle
È quasi sempre preferibile utilizzare effettivamente il cPicklemodulo piuttosto che pickleperché il primo è scritto in C ed è molto più veloce. Ci sono alcune sottili differenze tra loro, ma nella maggior parte delle situazioni sono equivalenti e la versione C fornirà prestazioni notevolmente superiori. Passare ad esso non potrebbe essere più semplice, basta cambiare l' importistruzione in questo:
import cPickle as pickle
In Python 3, è cPicklestato rinominato _pickle, ma farlo non è più necessario poiché il picklemodulo ora lo fa automaticamente: vedi Che differenza tra pickle e _pickle in python 3? .
Il riassunto è che potresti usare qualcosa di simile al seguente per assicurarti che il tuo codice userà sempre la versione C quando è disponibile in Python 2 e 3:
try:
import cPickle as pickle
except ModuleNotFoundError:
import pickle
Formati del flusso di dati (protocolli)
picklepuò leggere e scrivere file in diversi formati specifici di Python, chiamati protocolli come descritto nella documentazione , "Protocollo versione 0" è ASCII e quindi "leggibile dall'uomo". Le versioni> 0 sono binarie e la più alta disponibile dipende dalla versione di Python utilizzata. L'impostazione predefinita dipende anche dalla versione di Python. In Python 2 l'impostazione predefinita era Versione protocollo 0, ma in Python 3.8.1 è la versione Protocollo 4. In Python 3.x al modulo è stato pickle.DEFAULT_PROTOCOLaggiunto un modulo , ma ciò non esiste in Python 2.
Fortunatamente c'è una scorciatoia per scrivere pickle.HIGHEST_PROTOCOLin ogni chiamata (supponendo che sia quello che vuoi, e di solito lo fai), basta usare il numero letterale -1- simile a fare riferimento all'ultimo elemento di una sequenza tramite un indice negativo. Quindi, invece di scrivere:
pickle.dump(obj, output, pickle.HIGHEST_PROTOCOL)
Puoi semplicemente scrivere:
pickle.dump(obj, output, -1)
In entrambi i casi, avresti specificato il protocollo una sola volta se avessi creato un Pickleroggetto da utilizzare in più operazioni di pickle:
pickler = pickle.Pickler(output, -1)
pickler.dump(obj1)
pickler.dump(obj2)
etc...
Nota : se ti trovi in un ambiente con versioni diverse di Python, probabilmente vorrai utilizzare esplicitamente (ad esempio un hardcode) un numero di protocollo specifico che tutti possono leggere (le versioni successive possono generalmente leggere i file prodotti da quelli precedenti) .
Oggetti multipli
Mentre un file di salamoia può contenere qualsiasi numero di oggetti in salamoia, come mostrato nei campioni di cui sopra, quando c'è un numero imprecisato di loro, è spesso più facile per memorizzare tutti in una sorta di contenitore variabile dimensioni, come una list, tupleo dicte scrivere tutti al file in una sola chiamata:
tech_companies = [
Company('Apple', 114.18), Company('Google', 908.60), Company('Microsoft', 69.18)
]
save_object(tech_companies, 'tech_companies.pkl')
e ripristina l'elenco e tutto il resto in seguito con:
with open('tech_companies.pkl', 'rb') as input:
tech_companies = pickle.load(input)
Il vantaggio principale è che non è necessario sapere quante istanze di oggetto vengono salvate per poterle ricaricare in un secondo momento (anche se è possibile farlo senza tali informazioni , richiede un codice leggermente specializzato). Vedi le risposte alla domanda correlata Salvare e caricare più oggetti nel file pickle? per dettagli su diversi modi per farlo. Personalmente mi piace la risposta di @Lutz Prechelt al meglio. Ecco adattato agli esempi qui:
class Company:
def __init__(self, name, value):
self.name = name
self.value = value
def pickled_items(filename):
""" Unpickle a file of pickled data. """
with open(filename, "rb") as f:
while True:
try:
yield pickle.load(f)
except EOFError:
break
print('Companies in pickle file:')
for company in pickled_items('company_data.pkl'):
print(' name: {}, value: {}'.format(company.name, company.value))