Самый простой способ сохранить структуру данных в файл в 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. Он работает очень хорошо для вложенных диктовок и списков, но масштабируется и для более сложных структур данных (т. Е. С участием пользовательских объектов), и его большой плюс в том, что формат читабелен.

Другие вопросы по тегам