通過列表理解

Question

壓扁名單列表我試圖用壓平列表理解Python中的列表。我的名單有點像

[[1, 2, 3], [4, 5, 6], 7, 8] 

只是在列表名單打印，則單個項目我寫了這個代碼

def flat(listoflist): 
    for item in listoflist: 
      if type(item) != list: 
        print item 
      else: 
        for num in item: 
          print num 
>>> flat(list1) 
1 
2 
3 
4 
5 
6 
7 
8 

然後我用同樣的邏輯來拉平通過列表理解我的列表我得到出現以下錯誤

list2 = [item if type(item) != list else num for num in item for item in list1] 
    Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
    TypeError: 'int' object is not iterable 

如何使用列表理解來壓扁這種列表清單？

Answer 1

>>> from collections import Iterable 
>>> from itertools import chain 

一行代碼：

>>> list(chain.from_iterable(item if isinstance(item,Iterable) and 
        not isinstance(item, basestring) else [item] for item in lis)) 
[1, 2, 3, 4, 5, 6, 7, 8] 

可讀版本：

>>> def func(x):           #use `str` in py3.x 
...  if isinstance(x, Iterable) and not isinstance(x, basestring): 
...   return x 
...  return [x] 
... 
>>> list(chain.from_iterable(func(x) for x in lis)) 
[1, 2, 3, 4, 5, 6, 7, 8] 
#works for strings as well 
>>> lis = [[1, 2, 3], [4, 5, 6], 7, 8, "foobar"] 
>>> list(chain.from_iterable(func(x) for x in lis))                 
[1, 2, 3, 4, 5, 6, 7, 8, 'foobar'] 

使用嵌套列表理解：（去相比itertools.chain是緩慢）：

>>> [ele for item in (func(x) for x in lis) for ele in item] 
[1, 2, 3, 4, 5, 6, 7, 8, 'foobar'] 

Answer 2

你試圖通過一個號碼，你不能這樣做（因此錯誤）進行迭代。

如果你正在使用Python 2.7：

>>> from compiler.ast import flatten 
>>> flatten(l) 
[1, 2, 3, 4, 5, 6, 7, 8] 

但是千萬注意，模塊現在已經過時，並在Python 3不再存在

Answer 3

另一種解決方案使用發電機：

import collections 

def flatten(iterable): 
    for item in iterable: 
     if isinstance(item, collections.Iterable) and not isinstance(item, str): # `basestring` < 3.x 
      yield from item # `for subitem in item: yield item` < 3.3 
     else: 
      yield item 

>>> list(flatten([[1, 2, 3], [4, 5, 6], 7, 8])) 
[1, 2, 3, 4, 5, 6, 7, 8] 

Answer 4

沒有人已經考慮到通常的答案：

def flat(l): 
    return [y for x in l for y in x] 

這個問題有一些關於浮動在StackOverflo W上。