从嵌套列表中创建整数列表？

Question

我有一个嵌套列表，例如：

 test = [[15, [7, [None], [11, [None], [13, [None], [None]]]], [None]], [20, [None], [None]]] 

我想从中创建另一个列表，其中仅包含嵌套中的整数。哪个会返回这个：

[15, 7, 11, 13, 20]

我已经创建了这个递归函数来完成我需要完成的事情，但是，我不禁认为这不是最好的方法。有没有更pythonic或更有效的方法来做到这一点？

def nest_search(nest, hold=[]):
    for item in nest:
        if isinstance(item, int):
            hold.append(item)
        if isinstance(item, list):
            nest_search(item, hold)
    return hold

>>> print nest_search(test)
[15, 7, 11, 13, 20]

Answer 1

我看到的唯一不符合 Python 的就是默认参数。请参阅 this question 了解为什么这不会按您的预期方式工作。

这是我的解决方法：

def nest_search(nest, hold=None):
    if hold is None:
        hold = []
    for item in nest:
        if isinstance(item, int):
            hold.append(item)
        if isinstance(item, list):
            nest_search(item, hold)
    return hold

另一种实现方式是使函数成为生成器，它一个一个地产生值，而不是将它们添加到它最后返回的列表中。 （如果您确实需要一个列表，只需将生成器调用包装在 list 构造函数中即可。

def nest_search_gen(nest):
    for item in nest:
        if isinstance(item, int):
            yield item
        if isinstance(item, list):
            yield from nest_search_gen(item)

这使用了 Python 3.3 中引入的新 yield from 语法。如果您使用的是早期版本，您可以通过将最后一行替换为：

for i in nest_search_gen(item):
    yield i

Answer 2

使用flatten 发布的here 解决方案，您可以尝试以下方法。

>>> def flatten(x):
    try:
      it = iter(x)
    except TypeError:
      yield x
    else:
      for i in it:
        for j in flatten(i):
          yield j
>>> filter(bool, flatten(test))
[15, 7, 11, 13, 20]

我认为 flatten 和 filter 两个独立函数的使用更清晰，并且您鼓励模块化，允许一个单独使用另一个。