Question

目标：表达模拟表达式

{k1: v1, k2: v2 for k1, k2, v1, v2 in data}

或更具体的案例{k1: v, k2: v for k1, k2, v, _ in data}

只遍历data 1次（在给定的示例data中是4元组的可迭代对象）。

（以及关于列表理解的类似问题，例如[myfunc1(v1), myfunc2(v2) for v1, v2 in data]）。

我可以假设只使用自己的迭代器解决方案：

def getby(iterable_data, iterable_indexes):
    for indexed_data in iterable_data:
        for tupl in iterable_indexes:
            yield tuple(indexed_data[ind] for ind in tupl)
    raise StopIteration

工作示例：list(getby([('a', 'b', 'c'), ('d', 'e', 'f', 'g')], [(0, 1), (2, 1)]))返回[('a', 'b'), ('c', 'b'), ('d', 'e'), ('f', 'e')]。但可能它很慢。

我也可以根据itertools.chain和itertools.groupby来假设等效表达式如上所述：

list(chain.from_iterable(lst for lst, _ in groupby([('a', 'b', 'c'), ('d', 'e', 'f', 'g')],
                                                   lambda lst: [(lst[0], lst[1]), (lst[2], lst[1])])))

但可能它很难看。

基于ipython的两种解决方案的比较：第一个 1000个循环，最好是3个： 20.2 ms 每个循环，第二个 1000个循环，最好的3个循环： 4.12 ms 每个循环，所以第二个解决方案真的更快。但也许更优雅的解决方案存在？

Answer 1

理解格式严格限于每次迭代一个结果。您可以添加额外的循环，以循环您要插入的多个元素：

{k: v for k1, k2, v1, v2 in data for k, v in ((k1, v1), (k2, v2))}

或你的第二个例子：

{k: v for k1, k2, v, _ in data for k in (k1, k2)}

相当于：

result = {}
for k1, k2, v1, v2 in data:
    for k in (k1, k2):
        result[k] = v

如果你可以把它写成嵌套循环，可能在它们之间有if个语句，只需要在最里面的一个语句分配一个键值对，那么你可以用词典理解来表达它。

您也可以使用chain.from_iterable()执行此操作，但随后生成包含一系列键值对的列表或元组，并将结果传递给dict()函数：

dict(chain.from_iterable(((k1, v1), (k2, v2)) for k1, v1, k2, v2 in data))
dict(chain.from_iterable(((k1, v), (k2, v)) for k1, k2, v, _ in data))

Python：同时具有多个键值对的词典理解

1 个答案: