将嵌套列表转换为带有键和多个列表值（如果存在）的字典

Question

我有一个嵌套列表如下。 由于列表非常庞大并且为了减少搜索时间，认为转换为字典会更好。

以 innerlist[index] 为键，rest 为值。 Incase 键出现两次需要在字典中添加值列表。

列表

[
 ['001', 'xxxx', 'xxxx'], 
 ['002', '1', 'H', '0', 'H'], 
 ['002', '1', 'Z', '0', 'H']
]

需要Dict如下

   {
        '001': ['xxxx', 'xxxx'],
        '002': [
                ['1', 'H','0', 'H'],
                ['1', 'Z','0', 'H']
        ],
    }

我用下面的代码做到了，似乎仍然不是一个巨大列表的最佳方式。 同样对于非重复项目，我得到包含 1 个项目的嵌套列表。 '001': [['xxxx', 'xxxx']], 让我知道是否有更好的方法。

data_dict = dict()
for data in bytes_data:
    if data[0] in data_dict:
        data_dict[data[0]].append(data[1:])
    else:
        data_dict[data[0]] = [data[1:]]

Answer 1

如果您使用的是最新的 Python 版本（我认为是 3.5+），您可以更简洁地执行此操作，例如：

data = [
  ['001', 'xxxx', 'xxxx'],
  ['002', '1', 'H', '0', 'H'],
  ['002', '1', 'Z', '0', 'H']
]

data_dict = {}

for key, *vals in data:
    data_dict.setdefault(key, []).append(vals)

但我建议对其进行基准测试，您拥有的代码可能更有效，尤其是如果许多键在 src 列表中没有重复条目时。