提取列表元素

Question

d = {'a':[1,2,3], 'b':[4,5]} 

，我使用列表理解需要

[1,2,3,4,5] 

。我該怎麼做？

Answer 1

使用嵌套列表理解：

>>> [val for lst in d.values() for val in lst] 
[1, 2, 3, 4, 5] 

但是你可能需要先排序的字典（因爲類型的字典是無序的），以保證順序：使用

>>> [val for key in sorted(d) for val in d[key]] 
[1, 2, 3, 4, 5] 

Answer 2

容易和一個班輪解決方案列表理解：

>>> sorted([num for val in d.values() for num in val]) 
    [1, 2, 3, 4, 5] 

Answer 3

如果因爲每個臨時列表適用a = a + b

sum(d.values(), []) 

（如果需要添加sorted()。）

Answer 4

sum(d.values(),[])作品，但不是高性能：元素的順序並不重要，那麼你就不能打敗這個。

使用itertools.chain.from_iterable代替：

import itertools 
print(list(itertools.chain.from_iterable(d.values()))) 

或排序版本：

print(sorted(itertools.chain.from_iterable(d.values())))