Você pode usar o picklemódulo na biblioteca padrão. Aqui está uma aplicação elementar para o seu exemplo:
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
Você também pode definir seu próprio utilitário simples, como o seguinte, que abre um arquivo e grava um único objeto nele:
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')
Atualizar
Como essa é uma resposta tão popular, gostaria de abordar alguns tópicos de uso um pouco avançados.
cPickle(ou _pickle) vspickle
É quase sempre preferível usar o cPicklemódulo, em vez de pickleporque o primeiro é escrito em C e é muito mais rápido. Existem algumas diferenças sutis entre eles, mas na maioria das situações eles são equivalentes e a versão C fornecerá um desempenho muito superior. Mudar para ele não poderia ser mais fácil, basta alterar a importdeclaração para isso:
import cPickle as pickle
No Python 3, cPicklefoi renomeado _pickle, mas isso não é mais necessário, já que o picklemódulo agora o faz automaticamente - consulte Que diferença entre pickle e _pickle no python 3? .
O resumo é que você pode usar algo como o seguinte para garantir que seu código sempre use a versão C quando estiver disponível no Python 2 e 3:
try:
import cPickle as pickle
except ModuleNotFoundError:
import pickle
Formatos de fluxo de dados (protocolos)
picklepode ler e gravar arquivos em vários formatos diferentes, específicos para Python, chamados protocolos, conforme descrito na documentação , "Protocolo versão 0" é ASCII e, portanto, "legível por humanos". As versões> 0 são binárias e a mais alta disponível depende de qual versão do Python está sendo usada. O padrão também depende da versão do Python. No Python 2, o padrão era a versão Protocol 0, mas no Python 3.8.1, é a versão Protocol 4. No Python 3.x, o módulo foi pickle.DEFAULT_PROTOCOLadicionado, mas isso não existe no Python 2.
Felizmente, existe uma abreviação para escrever pickle.HIGHEST_PROTOCOLem todas as chamadas (assumindo que é isso que você deseja e costuma fazer), basta usar o número literal -1- semelhante a referenciar o último elemento de uma sequência por meio de um índice negativo. Então, em vez de escrever:
pickle.dump(obj, output, pickle.HIGHEST_PROTOCOL)
Você pode simplesmente escrever:
pickle.dump(obj, output, -1)
De qualquer forma, você só especificaria o protocolo uma vez se tivesse criado um Picklerobjeto para uso em várias operações de pickle:
pickler = pickle.Pickler(output, -1)
pickler.dump(obj1)
pickler.dump(obj2)
etc...
Nota : Se você estiver em um ambiente executando versões diferentes do Python, provavelmente desejará usar explicitamente (por exemplo, código rígido) um número de protocolo específico que todos possam ler (versões posteriores geralmente podem ler arquivos produzidos por versões anteriores) .
Vários Objetos
Enquanto um arquivo de picles pode conter qualquer número de objetos em conserva, como mostrado nas amostras acima, quando há um número desconhecido deles, muitas vezes é mais fácil de armazenar todos eles em algum tipo de recipiente de tamanho variável, como um list, tupleou dicte escrever todos eles para o arquivo em uma única chamada:
tech_companies = [
Company('Apple', 114.18), Company('Google', 908.60), Company('Microsoft', 69.18)
]
save_object(tech_companies, 'tech_companies.pkl')
e restaure a lista e tudo mais tarde com:
with open('tech_companies.pkl', 'rb') as input:
tech_companies = pickle.load(input)
A principal vantagem é que você não precisa saber quantas instâncias de objetos são salvas para carregá-las novamente mais tarde (embora fazer isso sem essas informações seja possível, isso requer algum código um pouco especializado). Veja as respostas para a pergunta relacionada Salvando e carregando vários objetos no arquivo pickle? para obter detalhes sobre diferentes maneiras de fazer isso. Pessoalmente, eu gosto mais da resposta de @Lutz Prechelt . Aqui está adaptado aos exemplos aqui:
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))