在Python中将字典保存到文件（替代pickle）？

Question

已回答 无论如何，我最后还是选择了pickle

好吧，根据我问的另一个问题的一些建议，我被告知使用pickle 将字典保存到文件中。

我试图保存到文件的字典是

members = {'Starspy' : 'SHSN4N', 'Test' : 'Test1'}

当 pickle 将其保存到文件时...这是格式

(dp0
S'Test'
p1
S'Test1'
p2
sS'Test2'
p3
S'Test2'
p4
sS'Starspy'
p5
S'SHSN4N'
p6
s.

您能给我另一种将字符串保存到文件的方法吗？

这是我希望它保存在

Members = {'Starspy' : 'SHSN4N', 'Test' : 'Test1'}

中的格式完整代码：

import sys
import shutil
import os
import pickle

tmp = os.path.isfile("members-tmp.pkl")
if tmp == True:
    os.remove("members-tmp.pkl")
shutil.copyfile("members.pkl", "members-tmp.pkl")

pkl_file = open('members-tmp.pkl', 'rb')
members = pickle.load(pkl_file)
pkl_file.close()

def show_menu():
    os.system("clear")
    print "\n","*" * 12, "MENU", "*" * 12
    print "1. List members"
    print "2. Add member"
    print "3. Delete member"
    print "99. Save"
    print "0. Abort"
    print "*" * 28, "\n"
    return input("Please make a selection: ")

def show_members(members):
    os.system("clear")
    print "\nNames", "     ", "Code"
    for keys in members.keys():
        print keys, " - ", members[keys]

def add_member(members):
    os.system("clear")
    name = raw_input("Please enter name: ")
    code = raw_input("Please enter code: ")
    members[name] = code
    output = open('members-tmp.pkl', 'wb')
    pickle.dump(members, output)
    output.close()
    return members


#with open("foo.txt", "a") as f:
#     f.write("new line\n")

running = 1

while running:
    selection = show_menu()
    if selection == 1:
        show_members(members)
        print "\n> " ,raw_input("Press enter to continue")
    elif selection == 2:
        members == add_member(members)
        print members
        print "\n> " ,raw_input("Press enter to continue")
    elif selection == 99:
        os.system("clear")
        shutil.copyfile("members-tmp.pkl", "members.pkl")
        print "Save Completed"
        print "\n> " ,raw_input("Press enter to continue")

    elif selection == 0:
        os.remove("members-tmp.pkl")
        sys.exit("Program Aborted")
    else:
        os.system("clear")
        print "That is not a valid option!"
        print "\n> " ,raw_input("Press enter to continue")

Answer 1

当然，将其保存为 CSV：

import csv
w = csv.writer(open("output.csv", "w"))
for key, val in dict.items():
    w.writerow([key, val])

然后读取它：

import csv
dict = {}
for key, val in csv.reader(open("input.csv")):
    dict[key] = val

另一种选择是 json（对于 2.6+ 版本，安装 json，对于 2.5 及以下版本，安装 simplejson）：

>>> import json
>>> dict = {"hello": "world"}
>>> json.dumps(dict)
'{"hello": "world"}'

Answer 2

目前最常见的序列化格式是 JSON，它受到普遍支持，并且非常清楚地表示简单的数据结构（例如字典）。

>>> members = {'Starspy' : 'SHSN4N', 'Test' : 'Test1'}
>>> json.dumps(members)
'{"Test": "Test1", "Starspy": "SHSN4N"}'
>>> json.loads(json.dumps(members))
{u'Test': u'Test1', u'Starspy': u'SHSN4N'}

Answer 3

YAML 格式（通过 pyyaml）可能是您的一个不错的选择：

http://en.wikipedia.org/维基/Yaml

http://pypi.python.org/pypi/PyYAML

Answer 4

尽管与 pp.pprint(the_dict) 不同，这不会那么漂亮，会一起运行，但 str() 至少可以以简单的方式保存字典对于快速任务：

f.write( str( the_dict ) )

Answer 5

您问了

我会尝试一下。如何指定将其转储到哪个文件/从中加载它？

除了写入字符串之外， json 模块还提供 < code>dump() 方法，写入文件：

>>> a = {'hello': 'world'}
>>> import json
>>> json.dump(a, file('filename.txt', 'w'))
>>> b = json.load(file('filename.txt'))
>>> b
{u'hello': u'world'}

还有一个 load() 方法用于读取。

Answer 6

虽然我建议使用 pickle，但如果您想要替代方案，则可以使用 klepto。

>>> init = {'y': 2, 'x': 1, 'z': 3}
>>> import klepto
>>> cache = klepto.archives.file_archive('memo', init, serialized=False)
>>> cache        
{'y': 2, 'x': 1, 'z': 3}
>>>
>>> # dump dictionary to the file 'memo.py'
>>> cache.dump() 
>>> 
>>> # import from 'memo.py'
>>> from memo import memo
>>> print memo
{'y': 2, 'x': 1, 'z': 3}

对于 klepto，如果您使用了 serialized=True，字典将被写入 memo.pkl 作为 pickled 字典，而不是使用clear文本。

您可以在此处获取 klepto：https://github.com/uqfoundation/klepto

dill 可能是比 pickle 本身更好的 pickle 选择，因为 dill 几乎可以序列化 python 中的任何内容。 klepto 也可以使用 dill。

您可以在此处获取 dill：https://github.com/uqfoundation/dill