Vous pouvez utiliser le picklemodule dans la bibliothèque standard. En voici une application élémentaire à votre exemple:
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
Vous pouvez également définir votre propre utilitaire simple comme le suivant qui ouvre un fichier et y écrit un seul objet:
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')
Mettre à jour
Puisqu'il s'agit d'une réponse si populaire, j'aimerais aborder quelques sujets d'utilisation légèrement avancés.
cPickle(ou _pickle) vspickle
Il est presque toujours préférable d'utiliser réellement le cPicklemodule plutôt que pickleparce que le premier est écrit en C et est beaucoup plus rapide. Il existe quelques différences subtiles entre eux, mais dans la plupart des situations, ils sont équivalents et la version C fournira des performances considérablement supérieures. Passer à cela ne pourrait pas être plus facile, changez simplement la importdéclaration en ceci:
import cPickle as pickle
En Python 3, a cPickleété renommé _pickle, mais cela n'est plus nécessaire puisque le picklemodule le fait maintenant automatiquement - voir Quelle différence entre pickle et _pickle en python 3? .
Le résumé est que vous pouvez utiliser quelque chose comme ce qui suit pour vous assurer que votre code utilisera toujours la version C lorsqu'elle est disponible à la fois en Python 2 et 3:
try:
import cPickle as pickle
except ModuleNotFoundError:
import pickle
Formats de flux de données (protocoles)
picklepeut lire et écrire des fichiers dans plusieurs formats différents, spécifiques à Python, appelés protocoles comme décrit dans la documentation , "Protocol version 0" est ASCII et donc "lisible par l'homme". Les versions> 0 sont binaires et la plus élevée disponible dépend de la version de Python utilisée. La valeur par défaut dépend également de la version de Python. Dans Python 2, la version par défaut était Protocol 0, mais dans Python 3.8.1, c'est la version Protocol 4. Dans Python 3.x, le module y avait été pickle.DEFAULT_PROTOCOLajouté, mais cela n'existe pas dans Python 2.
Heureusement, il y a un raccourci pour écrire pickle.HIGHEST_PROTOCOLdans chaque appel (en supposant que c'est ce que vous voulez, et vous le faites habituellement), utilisez simplement le numéro littéral -1- similaire à référencer le dernier élément d'une séquence via un index négatif. Donc, au lieu d'écrire:
pickle.dump(obj, output, pickle.HIGHEST_PROTOCOL)
Vous pouvez simplement écrire:
pickle.dump(obj, output, -1)
Quoi qu'il en soit, vous n'auriez à spécifier le protocole qu'une seule fois si vous avez créé un Picklerobjet à utiliser dans plusieurs opérations de décapage:
pickler = pickle.Pickler(output, -1)
pickler.dump(obj1)
pickler.dump(obj2)
etc...
Remarque : Si vous êtes dans un environnement exécutant différentes versions de Python, vous voudrez probablement utiliser explicitement (c'est-à-dire le code dur) un numéro de protocole spécifique que tous peuvent lire (les versions ultérieures peuvent généralement lire les fichiers produits par des versions antérieures) .
Objets multiples
Alors qu'un fichier cornichon peut contenir un certain nombre d'objets marinées, comme le montrent les exemples ci - dessus, quand il y a un nombre inconnu d'entre eux, il est souvent plus facile de les stocker dans une sorte de conteneur taille variable, comme un list, tupleou dictet écriture tous dans le fichier en un seul appel:
tech_companies = [
Company('Apple', 114.18), Company('Google', 908.60), Company('Microsoft', 69.18)
]
save_object(tech_companies, 'tech_companies.pkl')
et restaurer la liste et tout ce qu'il contient plus tard avec:
with open('tech_companies.pkl', 'rb') as input:
tech_companies = pickle.load(input)
L'avantage majeur est que vous n'avez pas besoin de savoir combien d'instances d'objet sont enregistrées pour les recharger plus tard (bien que cela sans ces informations soit possible, cela nécessite du code légèrement spécialisé). Voir les réponses à la question connexe Enregistrer et charger plusieurs objets dans un fichier pickle? pour plus de détails sur les différentes façons de procéder. Personnellement , je LIKE @Lutz Prechelt réponse le meilleur. Voici adapté aux exemples ici:
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))