Ви можете використовувати 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тому, що перший пишеться на С та набагато швидше. Існують деякі тонкі відмінності між ними, але в більшості ситуацій вони еквівалентні, а версія С забезпечить значно кращі показники. Перехід на нього не може бути простішим, просто змініть 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)
Основна перевага полягає в тому, що вам не потрібно знати, скільки об'єктів збережено, щоб потім їх знову завантажити (хоча це можливо і без цієї інформації , можливо, потрібен деякий трохи спеціалізований код). Дивіться відповіді на відповідне запитання Збереження та завантаження декількох об’єктів у файл маринування? для детальної інформації про різні способи зробити це. Особисто я , як @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))