Python从具有相同名称的字符串追加到列表

时间:2014-01-28 14:51:43

标签: python

我有以下列表。

('A', 'Steve', 'AAPLES', '0', '0', '15')
('B', 'Steve', 'ORANGES', '0', '0', '100')
('C', 'Paul', 'GRAPES', '0', '0', '500')

我想循环遍历每一行,然后将该行追加到与行[0]同名的列表中。

我正在尝试类似的事情:

A = []
B = []
C = []

for line in test_file:
    group = line[0].strip(' ')
    group.append(line)

这不起作用,因为'group'是一个字符串。我可以将组设为非字符串,以便我可以附加到该列表中吗?

1 个答案:

答案 0 :(得分:6)

使用将名称映射到列表的字典:

A = []
B = []
C = []

group_dict = {'A': A, 'B': B, 'C': C}

for line in test_file:
    group = line.strip()[0]
    group_dict[group].append(line)