如何拼合列表以返回包含所有元素的新列表？

Question

我正在尝试编写一个名为flatten_list的函数，该函数将可能嵌套的列表作为输入，并返回包含输入列表的所有元素的非嵌套列表。

我的代码：

def flatten_list(alist):
    """
    >>> flatten_list([1,2,3])
    [1, 2, 3]
    >>> flatten_list([1, [2,3], [4, 5], 6])
    [1, 2, 3, 4, 5, 6]
    """
    flat_list = []
    for element in alist:
        flat_list += element
    return flat_list

此代码适用于带有字符串的列表，但不适用于整数值。如何更改代码，使其同时适用于两者？

谢谢

Answer 1

通常，这将以递归方式进行，例如：

def flatten(input_, output=None):
    if output is None:
        output = []
    if isinstance(input_, basestring):
        output.append(input_)
    else:
        for item in input_:
            try:
                flatten(item, output)
            except TypeError:
                output.append(item)
    return output

这将可迭代的容器的任意组合（例如工作set，list，tuple，dict（仅按键））和内容（例如int，float，str），使用普通EAFP Python的风格。注意字符串的特定异常，您可能不想将其解压缩！

一些用法示例：

>>> flatten([1, [2, [3, [4, 5], 6], 7], 8])
[1, 2, 3, 4, 5, 6, 7, 8]
>>> flatten([1, "foo", ["bar", 2], 3])
[1, 'foo', 'bar', 2, 3]
>>> flatten([1, (2, 3), {4: 5}])
[1, 2, 3, 4]
>>> flatten("hello")
['hello']

非可迭代对象作为直接参数会发生什么：

>>> flatten(1)

Traceback (most recent call last):
  File "<pyshell#3>", line 1, in <module>
    flatten(1)
  File "<pyshell#1>", line 4, in flatten
    for item in input_:
TypeError: 'int' object is not iterable