Question

我们说我有两个由另一个函数生成的列表：

test = [[0, 1], [0, 2], [1, 5], [1,6], [2, 0], [3, 99], [3, 89], [3, 79]]
test2 = [[1, 4], [4, 1]]

我想将它们转换为关联数组以便快速查找，如下所示：

test: {0: [1, 2], 1: [5,6], 2: [0], 3: [99, 98, 97]}
test2: {1: [4], 4: [1]}

我可以这样做：

def list_to_dict(my_list):
    last_val = my_list[0][0]
    temp = []
    my_dict = {}

    for i in my_list:
        if last_val == i[0]:
            temp.append(i[1])
        else:
            #add the values to this key
            my_dict[last_val] = temp
            #reset the list
            temp = []
            temp.append(i[1])

        last_val = i[0]
    my_dict[last_val] = temp
    return my_dict

但是，这不是Pythonic。是否有更多的Pythonic方法来实现这一目标？

Answer 1

使用collections.defaultdict：

>>> test = [[0, 1], [0, 2], [1, 5], [1,6], [2, 0], [3, 99], [3, 89], [3, 79]]
>>> 
>>> from collections import defaultdict
>>> d = defaultdict(list)
>>> 
>>> for i, j in test:
...     d[i].append(j)
... 
>>> d
defaultdict(<type 'list'>, {0: [1, 2], 1: [5, 6], 2: [0], 3: [99, 89, 79]})

Answer 2

您也可以通过字典分组来完成此操作，如下所示：

    def list_to_dict(my_list):
        d = {}

        for pair in my_list:
            first = pair[0]
            second = pair[1]
            if first not in d:
                d[first] = []
            d[first].append(second)
        return d

如果密钥不可用，您可以使用setdefault（）：

    def list_to_dict(my_list):
        d = {}
        for pair in my_list:
            d.setdefault(pair[0], []).append(pair[1])
        return d

    # Output 

    {0: [1, 2], 1: [5, 6], 2: [0], 3: [99, 89, 79]}

如果你愿意，可以很容易地操作这些来使用defaultdict。

Answer 3

这似乎是itertools.groupby的完美用例：

test = [[0, 1], [0, 2], [1, 5], [1,6], [2, 0], [3, 99], [3, 89], [3, 79]]

import itertools
d = {k:[t[1] for t in g] for k,g in itertools.groupby(test, key=lambda t: t[0])}

print(d)

请注意，groupby适用于iterables，默认情况下它返回grouper个对象。结果是：

{0: [1, 2], 1: [5, 6], 2: [0], 3: [99, 89, 79]}

如何将包含重复值的列表转换为以列表作为值的字典？

3 个答案: