![](/img/trans.png)
[英]Python list comprehension for list of dictd into a list of tuple with a subset of dict key-value pairs
[英]Python: dict comprehension with multiple key-value pairs in the same time
目標:表達模擬
{k1: v1, k2: v2 for k1, k2, v1, v2 in data}
或更具體的情況{k1: v, k2: v for k1, k2, v, _ in data}
只迭代data
一次 (在給定的示例中, 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每循環 ,所以第二個解決方案真的更快。 但也許更優雅的解決方案存在?
理解格式嚴格限於每次迭代一次結果。 您可以添加額外的循環,以循環您要插入的多個元素:
{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))
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.