Puede usar el picklemódulo en la biblioteca estándar. Aquí hay una aplicación básica para su ejemplo:
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
También puede definir su propia utilidad simple como la siguiente que abre un archivo y escribe un solo objeto en él:
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')
Actualizar
Dado que esta es una respuesta tan popular, me gustaría tocar algunos temas de uso ligeramente avanzados.
cPickle(o _pickle) vspickle
Casi siempre es preferible usar el cPicklemódulo en lugar de hacerlo pickleporque el primero está escrito en C y es mucho más rápido. Hay algunas diferencias sutiles entre ellos, pero en la mayoría de las situaciones son equivalentes y la versión C proporcionará un rendimiento muy superior. Cambiar a él no podría ser más fácil, solo cambie la importdeclaración a esto:
import cPickle as pickle
En Python 3, cPicklese le cambió el nombre _pickle, pero hacerlo ya no es necesario ya que el picklemódulo ahora lo hace automáticamente. ¿Qué diferencia hay entre pickle y _pickle en python 3? .
El resumen es que podría usar algo como lo siguiente para asegurarse de que su código siempre use la versión C cuando esté disponible en Python 2 y 3:
try:
import cPickle as pickle
except ModuleNotFoundError:
import pickle
Formatos de flujo de datos (protocolos)
picklepuede leer y escribir archivos en varios formatos diferentes, específicos de Python, llamados protocolos como se describe en la documentación , "Protocolo versión 0" es ASCII y por lo tanto "legible por humanos". Las versiones> 0 son binarias y la más alta disponible depende de la versión de Python que se esté utilizando. El valor predeterminado también depende de la versión de Python. En Python 2, el valor predeterminado era la versión del Protocolo 0, pero en Python 3.8.1, es la versión del Protocolo 4. En Python 3.x, el módulo tenía un pickle.DEFAULT_PROTOCOLagregado, pero eso no existe en Python 2.
Afortunadamente, hay una forma abreviada de escribir pickle.HIGHEST_PROTOCOLen cada llamada (suponiendo que eso es lo que desea, y generalmente lo hace), solo use el número literal -1, similar a hacer referencia al último elemento de una secuencia a través de un índice negativo. Entonces, en lugar de escribir:
pickle.dump(obj, output, pickle.HIGHEST_PROTOCOL)
Solo puedes escribir:
pickle.dump(obj, output, -1)
De cualquier manera, solo habría especificado el protocolo una vez si hubiera creado un Picklerobjeto para usar en múltiples operaciones de encurtido:
pickler = pickle.Pickler(output, -1)
pickler.dump(obj1)
pickler.dump(obj2)
etc...
Nota : Si se encuentra en un entorno que ejecuta diferentes versiones de Python, entonces probablemente quiera usar explícitamente (es decir, código rígido) un número de protocolo específico que todos puedan leer (las versiones posteriores generalmente pueden leer archivos producidos por versiones anteriores) .
Objetos múltiples
Mientras que un archivo de salmuera puede contener cualquier número de objetos en escabeche, como se muestra en los ejemplos anteriores, cuando hay un número desconocido de ellos, a menudo es más fácil de almacenar todos ellos en algún tipo de recipiente de tamaño variable, como una list, tupleo dictya escribir todos ellos al archivo en una sola llamada:
tech_companies = [
Company('Apple', 114.18), Company('Google', 908.60), Company('Microsoft', 69.18)
]
save_object(tech_companies, 'tech_companies.pkl')
y restaure la lista y todo en ella más tarde con:
with open('tech_companies.pkl', 'rb') as input:
tech_companies = pickle.load(input)
La principal ventaja es que no necesita saber cuántas instancias de objetos se guardan para cargarlas de nuevo más tarde (aunque hacerlo sin esa información es posible, requiere un código ligeramente especializado). Ver las respuestas a la pregunta relacionada ¿ Guardar y cargar varios objetos en un archivo pickle? para obtener detalles sobre las diferentes formas de hacer esto. Personalmente, me gusta la respuesta de @Lutz Prechelt . Aquí está adaptado a los ejemplos aquí:
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))