将两个嵌套列表压缩到单个级别字典的最有效方法是什么

Question

例如:

list1=['k1','k2','k3',['k4','k5',['k6','k7']]]
list2=['v1','v2','v3',['v4','v5',['v6','v7']]]

我想将它们合并到这样的字典中:

dict1={'k1':'v1','k2':'v2','k3':'v3','k4':'v4','k5':'v5','k6':'v6','k7':'v7'}

我有办法做到这一点,但我认为这需要太多时间:

def mergeToDict(keyList, valueList):
    resultDict = {}
    for key, value in itertools.izip(keyList, valueList):
        if type(key) == list and type(value) == list:
            resultDict=dict(resultDict,**mergeToDict(key, value))
        elif type(key) != list and type(key) != dict and type(key) != tuple:
            resultDict[key] = value
    return resultDict

有没有更好的想法？

Answer 1

我会使用某种扁平功能:

def flatten(it):
    if isinstance(it, str):
        yield it
        return
    try:
        for x in it:
            for y in flatten(x):
                yield y
    except TypeError:
        yield it

现在你可以做到

from itertools import izip
my_dict = dict(izip(flatten(list1), flatten(list2)))

我认为这种方式对于读者来说更通用,更透明.

Answer 2

我认为如果您只有像您提出的用例（嵌套列表但具有相同形状）这样的用例，那么您根本不需要展平。这是一种方法，至少在我的机器上比你的机器快 2-3 倍（同样只适用于该约束）：

def appendDict(list1, list2, resultDict):
    for idx, val in enumerate(list1):
        if isinstance(val, list):       
            appendDict(val, list2[idx], resultDict)
        else:
            resultDict[val] = list2[idx]

list1=['k1','k2','k3',['k4','k5',['k6','k7']]]
list2=['v1','v2','v3',['v4','v5',['v6','v7']]]
resultDict = {}
appendDict(list1, list2, resultDict)
print resultDict

{'k3': 'v3', 'k2': 'v2', 'k1': 'v1', 'k7': 'v7', 'k6': 'v6', 'k5': 'v5', 'k4': 'v4'}

以及方法的比较：

OP 的方法，运行 10000 次：0.290050983429

其他建议的方法，10000 次运行：0.580717086792

此方法运行 10000 次：0.155267000198

也许不如其他解决方案那么优雅，但性能似乎是这里的主要问题。