Самый простой способ сохранить структуру данных в файл в Python?
Допустим, у меня есть что-то вроде этого:
d = { "abc" : [1, 2, 3], "qwerty" : [4,5,6] }
Какой самый простой способ программно получить это в файл, который я могу загрузить из Python позже?
Могу ли я как-то сохранить его как исходный код Python (из сценария Python, а не вручную!), Затем import
это позже?
Или я должен использовать JSON или что-то?
7 ответов
Используйте модуль рассола.
import pickle
d = { "abc" : [1, 2, 3], "qwerty" : [4,5,6] }
afile = open(r'C:\d.pkl', 'wb')
pickle.dump(d, afile)
afile.close()
#reload object from file
file2 = open(r'C:\d.pkl', 'rb')
new_d = pickle.load(file2)
file2.close()
#print dictionary object loaded from file
print new_d
Выберите: Стандартная библиотека Python - Устойчивость данных. Какой из них наиболее подходит, может варьироваться в зависимости от ваших конкретных потребностей.
pickle
это, пожалуй, самый простой и наиболее эффективный способ "записать произвольный объект в файл и восстановить его" - он может автоматически обрабатывать пользовательские классы и циклические ссылки.
Для лучшей производительности травления (скорость и пространство) используйте cPickle
в HIGHEST_PROTOCOL
,
Попробуйте модуль shelve, который даст вам постоянный словарь, например:
import shelve
d = { "abc" : [1, 2, 3], "qwerty" : [4,5,6] }
shelf = shelve.open('shelf_file')
for key in d:
shelf[key] = d[key]
shelf.close()
....
# reopen the shelf
shelf = shelve.open('shelf_file')
print(shelf) # => {'qwerty': [4, 5, 6], 'abc': [1, 2, 3]}
У JSON есть недостатки, но когда он отвечает вашим потребностям, он также:
- прост в использовании
- включены в стандартную библиотеку как
json
модуль - интерфейс чем-то похож на
pickle
, который может справиться с более сложными ситуациями - редактируемый человеком текст для отладки, обмена и контроля версий
- действительный код Python
- хорошо зарекомендовавший себя в Интернете (если ваша программа касается какого-либо домена)
Вы также можете захотеть взглянуть на объектную базу данных Zope, тем сложнее вы получите:-) Вероятно, излишне для того, что у вас есть, но она хорошо масштабируется и не слишком сложна в использовании.
Если вы хотите сохранить его в удобном для чтения формате JSON, используйте repr
сериализовать объект и eval
десериализовать это.
repr(object) -> string
Вернуть каноническое строковое представление объекта. Для большинства типов объектов
eval(repr(object)) == object
,
Чтобы добавить к предыдущим предложениям, если вы хотите, чтобы формат файла был легко читаемым и изменяемым, вы также можете использовать YAML. Он работает очень хорошо для вложенных диктовок и списков, но масштабируется и для более сложных структур данных (т. Е. С участием пользовательских объектов), и его большой плюс в том, что формат читабелен.