目标:表达的工作模拟

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

或更多特殊情况{k1: v, k2: v for k1, k2, v, _ in data}
它只遍历data1次(在给定的示例中data是4元组的iterable对象)。
(以及关于列表理解的类似问题,例如[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.chainitertools.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.2ms/循环,后1000个循环,最好的3:4.12ms/循环,所以第二种解决方案确实更快。但也许是更优雅的解决方案存在吗?

最佳答案

理解格式严格限制为每次迭代一个结果。但是,您可以添加额外的循环,以循环要插入的多个元素:

{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))

10-06 10:12
查看更多