0

我正在尝试编写一个函数，它将我的文本文件变成带有子集的字典。到目前为止我加载的文本文档显示如下：

101
102
103
201, John Cleese, 5/5/12, 5/7/12
202
203, Eric Idle, 7/5/12, 8/7/12
301
302
303

该函数的结果应该加载信息并将其返回为：

[('101', None), ('102', None), ('103', None),
('201', Guest(John Cleese, 05/05/12, 05/07/12)), ('202', None),
('203', Guest(Eric Idle, 07/05/12, 08/07/12)), ('301', None),
('302', None), ('303', None)]

我一直在尝试 isdigit 但无济于事。我设法让最后一个房间（303）几乎可以正常工作。

def load_rooms(self, filename):
    fd = open(filename, 'rU')
    self._rooms = {}
    for i in fd:
        i = i.rstrip()
        print i

    if i.isdigit():
        self._rooms[i] = None
        print self._rooms
    else:
        print i

显示

101
102
103
201, John Cleese, 5/5/12, 5/7/12
202
203, Eric Idle, 7/5/12, 8/7/12
301
302
303
{'303': None}

哦，哇，我刚刚注意到房间 303 显示了两次......我的问题是，我将如何加载文本文件？我是否必须确保所有代码都采用格式，或者我可以只编写一个函数将其转换为带有子集的字典吗？我对字典很陌生，所以一点帮助会很棒。谢谢

4

2 回答 2

1 
import re

class Guest(object):
  ...

data = []
with open('data.txt') as f:
  for line in f:
    tok = re.split(r'\s*,\s*', line.strip())
    if len(tok) > 1:
      guest = Guest(tok[1], tok[2], tok[3])
    else:
      guest = None
    data.append((tok[0], guest))

print(data)

实现Guest类留给读者作为练习。

于 2012-05-15T12:39:07.063 回答
0

我不确定您是否需要该re模块

尝试这个

def load_file( filename ):
    text = open( "data.txt" ).readlines()
    rooms = {}
    for line in text:
        room_data = line.strip( '\n' ).split( ',' )
        if len( room_data ) > 1:
            #rooms[room_data[0]] = room_data[1:]
            rooms[room_data[0]] = tuple( room_data[1:] )
        else:
            rooms[room_data[0]] = None

    return rooms

您在标题中提到了字典，但您的预期结果是嵌套列表。

编辑

回答了吗？

于 2012-05-15T15:39:03.977 回答