将字典保存到Python中的文件(pickle的替代方案)?

2024-09-15 17:40:02 发布

您现在位置:Python中文网/ 问答频道 /正文

回答说我最后还是吃了泡菜

好吧,我问了另一个问题,有人建议我用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.

你能给我另一种方法把字符串保存到文件中吗?

这是我想保存的格式

成员={'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")

Tags: 文件importinputrawosshowsystemtmp
3条回答

目前最常见的序列化格式是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'}

当然,保存为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"}'

YAML格式(通过pyyaml)可能是一个不错的选择:

http://en.wikipedia.org/wiki/Yaml

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

相关问题 更多 >