将相似的列表元素分组到 dict 中，而不将其多次出现

Question

我有一个巨大的列表，我将它转换为一个字典（基于第一个列表元素），以便使用键进行快速搜索。

列表：

[0100,A,1.00,1]
.
.
[0450,A,1.00,1]
[0470,B,1.00,1]
[0480,A,1.00,1]
[0490,A,1.00,1]
[0500,A,1.00,1] #list-1 for below ref
[0510,B,1.00,1]
[0520,A,1.00,1]
[0530,A,1.00,1]
[0500,A,1.00,1] #list-2 for below ref
[0510,B,1.00,1]
[0520,X,1.00,1] #........

转换为 Dic：

for key, *vals in bytes_data: #Probably need a diff approach here itself instead appending
    data_dict.setdefault(key, []).append(vals) 

字典看起来像

{
    '0450': [[A,1.00,1]],
    '0470': [[B,1.00,1]],
    '0480': [[A,1.00,1]], #......
}

现在，我当前的场景需要根据情况对 4xx/5xx/... 等数据进行分块。 为此，我使用..

key_series = ["0" + str(x) for x in range(500, 600, 10)]
article_data = {
    key: data_dict[key] for key in set(key_series) & set(data_dict)
}

问题是，对于像 5xx 这样的某些系列，会出现多次。 在这种情况下，我的 dict 被分组为

{
    0500: [list-1,list-2,..],
    0510: [list-1,list-2,..]
}

但是，我需要类似的东西

{
0500-1: {0500: [list-1], 0510: [list-1],....}, 
0500-2: {0500: [list-2], 0510: [list-2],....},
}

有什么技巧可以做到这一点？ 谢谢。

Answer 1

不确定，如果这是您想要的，请告诉我这是否解决了您的问题

from collections import defaultdict
data_dict = {
    "0500": [["A",1.00,1]],
    "0510": [["A",1.00,1], ["B",1.00,1], ["B",1.00,1]],
    "0520": [["A",1.00,1], ["D",1.00,1]]
}
key_series = ["0" + str(x) for x in range(500, 600, 10)]
article_data = {
    key: data_dict[key] for key in set(key_series) & set(data_dict)
}
res = defaultdict(dict)
for k ,v in data_dict.items():
    for i, d in enumerate(v):
        # for now 0500 is hardcoded, which can be made dynamic as per requirement
        res[f"0500-{i+1}"][k] = d

print(res)