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

Question

我有一个“银行应用程序类型”“银行名称”及其本地ID值的列表。像这样：

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

我想根据他们的银行名称对他们的身份证进行分类，所以我为此使用了一个简化的列表。

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])

但从这一点上说，我无法实现我想要的。我试图迭代分组的值，但没有成功。

在这一点上，我怎样才能达到所需的结果呢？

Answer 1

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]}

Answer 2

这是一个没有依赖项的一行代码：

{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中为每个组找到值。