根据单个值对元素进行分组

时间:2013-09-16 10:41:21

标签: python list grouping

我有一个“银行申请类型”“银行名称”列表及其本地ID值。像:

BANKS = [
    (1, "Bank A Web", "BankA"),
    (2, "Bank B ATM", "BankB"),
    (3, "Bank A Wap", "BankA"),    
    (4, "Bank C",     "BankC"),
]

我想根据他们的银行名称对他们的ID进行分类,所以我为此目的使用了简化列表

BANKS_CATS = [
    (1, "BankA"),
    (2, "BankB"),
    (3, "BankA"), 
    (4, "BankC"),
]

我希望将每个银行ID与其他相关银行ID(来自同一银行的不同服务)进行匹配

{1: [1, 3],
 2: [2],
 3: [1, 3],
 4: [4]}

此值列表也是一个简化示例,我的实际银行名称和银行服务名称不同,并且没有相似的模式可供匹配以对其进行分类。

我订购了我的实际清单

my_new_list = sorted(BANKS_CATS, key=lambda s: s[1])

然后尝试将其分组

from itertools import groupby
grouped = groupby(my_new_list, key=lambda s: s[1])

但形成了这一点,我无法达到我想要的效果。我尝试迭代分组值而没有成功。

如何从这一点获得所需的结果?

2 个答案:

答案 0 :(得分:2)

my_list = [
    (1, "BankA"),
    (2, "BankB"),
    (3, "BankA"), 
    (4, "BankC"),
]

ids, result = {}, {}
for num, bank in my_list: ids.setdefault(bank, []).append (num)
for num, bank in my_list: result[num] = ids[bank]
print result

<强>输出

{1: [1, 3], 2: [2], 3: [1, 3], 4: [4]}

答案 1 :(得分:0)

这是一个没有依赖关系的单行代码:

{int(t[1][4:]):[i[0] for i in my_list if i[1] == t[1]] for t in my_list}

首先从my_list收集组,然后在my_list中为每个组找到值。