130

我正在尝试将字典写入 txt 文件。然后通过键入带有 的键来读取 dict 值raw_input。我觉得我只是错过了一步,但我已经找了一段时间了。

我收到这个错误

File "name.py", line 24, in reading
    print whip[name]
TypeError: string indices must be integers, not str

我的代码:

#!/usr/bin/env python
from sys import exit

class Person(object):
    def __init__(self):
        self.name = ""
        self.address = ""
        self.phone = ""
        self.age = ""
        self.whip = {}

    def writing(self):
        self.whip[p.name] = p.age, p.address, p.phone
        target = open('deed.txt', 'a')
        target.write(str(self.whip))
        print self.whip

    def reading(self):
        self.whip = open('deed.txt', 'r').read()
        name = raw_input("> ")
        if name in self.whip:
            print self.whip[name]

p = Person()

while True:
    print "Type:\n\t*read to read data base\n\t*write to write to data base\n\t*exit to exit"
    action = raw_input("\n> ")
    if "write" in action:
        p.name = raw_input("Name?\n> ")
        p.phone = raw_input("Phone Number?\n> ")
        p.age = raw_input("Age?\n> ")
        p.address = raw_input("Address?\n>")
        p.writing()
    elif "read" in action:
        p.reading()
    elif "exit" in action:
        exit(0)
4

6 回答 6

183

您是否尝试过 json 模块?JSON 格式与 python 字典非常相似。它是人类可读/可写的:

>>> import json
>>> d = {"one":1, "two":2}
>>> json.dump(d, open("text.txt",'w'))

此代码转储到文本文件

$ cat text.txt 
{"two": 2, "one": 1}

您也可以从 JSON 文件加载:

>>> d2 = json.load(open("text.txt"))
>>> print d2
{u'two': 2, u'one': 1}
于 2012-06-14T05:14:27.987 回答
107

您的代码几乎是正确的!你是对的,你只是少了一步。当您读入文件时,您将其作为字符串读取;但是您想将字符串转回字典。

您看到的错误消息是因为self.whip是字符串,而不是字典。因此,您需要将字符串转换为字典。

例子

这是最简单的方法:将字符串输入eval(). 像这样:

def reading(self):
    s = open('deed.txt', 'r').read()
    self.whip = eval(s)

您可以在一行中完成,但我认为这样看起来很乱:

def reading(self):
    self.whip = eval(open('deed.txt', 'r').read())

eval()有时不推荐。问题是它eval()会评估任何字符串,如果有人欺骗你运行一个非常棘手的字符串,可能会发生一些不好的事情。在这种情况下,您只是eval()在自己的文件上运行,所以应该没问题。

但是因为eval()有用,所以有人做了一个更安全的替代品。这被调用literal_eval,你可以从一个名为ast.

import ast

def reading(self):
    s = open('deed.txt', 'r').read()
    self.whip = ast.literal_eval(s)

ast.literal_eval()只会评估变成基本 Python 类型的字符串,所以一个棘手的字符串不可能在你的计算机上做坏事。

编辑

实际上,Python 中的最佳实践是使用with语句来确保文件正确关闭。重写上面的语句以使用with语句:

import ast

def reading(self):
    with open('deed.txt', 'r') as f:
        s = f.read()
        self.whip = ast.literal_eval(s)

在最流行的 Python 中,称为“CPython”,您通常不需要该with语句,因为内置的“垃圾收集”功能会确定您已完成文件并为您关闭它。但是其他 Python 实现,例如“Jython”(用于 Java VM 的 Python)或“PyPy”(具有即时代码优化的非常酷的实验系统)可能无法为您关闭文件。养成使用的习惯是件好事with,而且我认为它使代码很容易理解。

于 2012-06-14T05:20:12.273 回答
75

要将 Python 对象存储在文件中,请使用以下pickle模块:

import pickle

a = {
  'a': 1,
  'b': 2
}

with open('file.txt', 'wb') as handle:
  pickle.dump(a, handle)

with open('file.txt', 'rb') as handle:
  b = pickle.loads(handle.read())

print a == b # True

请注意,我从未设置过b = a,而是腌制a到一个文件,然后将其解压缩到b.

至于你的错误:

self.whip = open('deed.txt', 'r').read()

self.whip是一个字典对象。deed.txt包含文本,因此当您加载deed.txtinto的内容时self.whipself.whip将成为其自身的字符串表示形式。

您可能希望将字符串评估回 Python 对象:

self.whip = eval(open('deed.txt', 'r').read())

注意eval听起来如何evil。那是故意的。请改用该pickle模块。

于 2012-06-14T05:14:09.857 回答
9

您好,有一种方法可以将字典写入和读取到文件中,您可以将字典转换为 JSON 格式并快速读取和写入,只需执行以下操作:

写下你的日期:

 import json

 your_dictionary = {"some_date" : "date"}
 f = open('destFile.txt', 'w+')
 f.write(json.dumps(your_dictionary))

读取您的数据:

 import json

 f = open('destFile.txt', 'r')
 your_dictionary = json.loads(f.read())
于 2017-08-12T05:37:20.037 回答
8

我创建了自己的功能,这些功能非常好用:

def writeDict(dict, filename, sep):
    with open(filename, "a") as f:
        for i in dict.keys():            
            f.write(i + " " + sep.join([str(x) for x in dict[i]]) + "\n")

它将首先存储键名,然后是所有值。请注意,在这种情况下,我的 dict 包含整数,这就是它转换为int. 这很可能是您需要根据自己的情况进行更改的部分。

def readDict(filename, sep):
    with open(filename, "r") as f:
        dict = {}
        for line in f:
            values = line.split(sep)
            dict[values[0]] = {int(x) for x in values[1:len(values)]}
        return(dict)
于 2013-09-15T08:27:46.853 回答
6

您可以遍历键值对并将其写入文件

pair = {'name': name,'location': location}
with open('F:\\twitter.json', 'a') as f:
     f.writelines('{}:{}'.format(k,v) for k, v in pair.items())
     f.write('\n')
于 2015-12-09T01:50:52.290 回答