Вы можете использовать pickleмодуль в стандартной библиотеке. Вот элементарное применение этого к вашему примеру:
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
Вы также можете определить свою собственную простую утилиту, например, которая открывает файл и записывает в него один объект:
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')
Обновить
Поскольку это такой популярный ответ, я бы хотел коснуться нескольких немного продвинутых тем использования.
cPickle(или _pickle) противpickle
Практически всегда предпочтительнее использовать cPickleмодуль, чем pickleтот, который написан на C и работает намного быстрее. Между ними есть некоторые тонкие различия, но в большинстве случаев они эквивалентны, и версия C обеспечит значительно более высокую производительность. Переход на это не может быть проще, просто измените importутверждение на это:
import cPickle as pickle
В Python 3 cPickleбыл переименован _pickle, но делать это больше не нужно, поскольку pickleмодуль теперь делает это автоматически - см. Какая разница между pickle и _pickle в python 3? ,
Вкратце вы можете использовать что-то вроде следующего, чтобы гарантировать, что ваш код всегда будет использовать версию C, когда она доступна как в Python 2, так и в 3:
try:
import cPickle as pickle
except ModuleNotFoundError:
import pickle
Форматы потока данных (протоколы)
pickleможет читать и записывать файлы в нескольких различных, специфичных для Python форматах, которые называются протоколами, как описано в документации. «Протокол версии 0» является ASCII и, следовательно, «читаемым человеком». Версии> 0 являются двоичными, и максимальная доступная версия зависит от того, какая версия Python используется. Значение по умолчанию также зависит от версии Python. В Python 2 по умолчанию была версия протокола 0, но в Python 3.8.1 это версия протокола 4. В Python 3.x модуль был pickle.DEFAULT_PROTOCOLдобавлен к нему, но его нет в Python 2.
К счастью, есть запись pickle.HIGHEST_PROTOCOLдля каждого вызова (при условии, что это то, что вы хотите, и вы обычно это делаете), просто используйте буквальный номер -1- аналогично ссылке на последний элемент последовательности через отрицательный индекс. Итак, вместо того, чтобы писать:
pickle.dump(obj, output, pickle.HIGHEST_PROTOCOL)
Вы можете просто написать:
pickle.dump(obj, output, -1)
В любом случае, вам нужно было бы указать протокол только один раз, если бы вы создали Picklerобъект для использования в нескольких операциях рассола:
pickler = pickle.Pickler(output, -1)
pickler.dump(obj1)
pickler.dump(obj2)
etc...
Примечание . Если вы работаете в среде, в которой работают разные версии Python, вам, вероятно, захочется явно использовать (т. Е. Использовать жесткий код) конкретный номер протокола, который все они могут прочитать (более поздние версии обычно могут читать файлы, созданные более ранними версиями) ,
Несколько объектов
В то время как рассол файл может содержать любое количество соленых объектов, как показано в примерах выше, когда есть неизвестное число из них, часто проще хранить их в каком - то контейнер переменно размера, подобно list, tupleили dictи писать все они в файл за один вызов:
tech_companies = [
Company('Apple', 114.18), Company('Google', 908.60), Company('Microsoft', 69.18)
]
save_object(tech_companies, 'tech_companies.pkl')
и восстановить список и все в нем позже с помощью:
with open('tech_companies.pkl', 'rb') as input:
tech_companies = pickle.load(input)
Главное преимущество является то, что вам не нужно знать , сколько объекта экземпляров сохраняются для того , чтобы загрузить их позже (хотя это и без этой информации является возможным, требует немного специализированного кода). См. Ответы на связанный вопрос. Сохранение и загрузка нескольких объектов в файл pickle? для деталей о различных способах сделать это. Лично мне больше всего нравится ответ @Lutz Prechelt . Вот это адаптировано к примерам здесь:
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))