在python中合並列表列表

Question

我想合並列表列表，這是下面的示例

mylist = [[['a', 'b'], ['c', 'd']],
          [['e', 'f'], ['g', 'h']]]

輸出應為：

output = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']

我也嘗試使用itertools ，但這是它返回的內容

>>> combined = list(itertools.chain.from_iterable(mylist))
>>> combined
>>> [['a', 'b'], ['c', 'd'], ['e', 'f'], ['g', 'h']]

我如何實現這一目標['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']

誰能強調我所缺少的東西？

Answer 1

itertools方法不起作用的原因是因為您擁有的不是列表列表，而是列表列表。 itertools工作正常，只需將列表弄平一次。 使用部分展平的列表作為參數再次調用完全相同的函數將起作用：

flat = list(itertools.chain.from_iterable(itertools.chain.from_iterable(mylist)))

或者，一個簡單的列表理解解決方案：

flat = [item for slist in mylist for sslist in slist for item in sslist]

這基本上轉化為：

for slist in mylist:
    for sslist in slist:
        for item in sslist:
            flat.append(item)

請記住，這兩種解決方案僅適用於處理雙重嵌套。 如果有可能您將不得不處理更多的嵌套，建議您查找如何展平任意嵌套列表。

Answer 2

正如其他人所指出的，您這里有兩個級別，因此需要兩個調用chain 。 但是您實際上並不需要from_iterable調用； 您可以改用*語法：

list(itertools.chain(*itertools.chain(*mylist)))

Answer 3

import numpy as np

mylist = [ [['a', 'b'], ['c', 'd']], [['e', 'f'], ['g', 'h']] ]    
a = np.array(mylist).flatten().tolist()

print(a)

輸出：

['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']