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

Question

我有一个“银行应用程序类型”“银行名称”及其本地 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]}

此外，该值列表是一个简化的示例，我的实际银行名称和银行服务名称不同，并且没有类似的模式来匹配对它们进行分类。

我订购了我的实际 list

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