展平不规则（任意嵌套）列表

我的解决方案：

import collections


def flatten(x):
    if isinstance(x, collections.Iterable):
        return [a for i in x for a in flatten(i)]
    else:
        return [x]

更简洁一点，但几乎相同。

使用生成器函数可以使您的示例更易于阅读并提高性能。

Python 2 中文文档

使用 2.6 中添加的可迭代 ABC：

from collections import Iterable

def flatten(xs):
    for x in xs:
        if isinstance(x, Iterable) and not isinstance(x, basestring):
            for item in flatten(x):
                yield item
        else:
            yield x

Python 3 （英语）

在 Python 3 中，不再是，但元组给出了相同的效果。此外，运算符的产量一次从生成器返回一个项目。basestring(str, bytes)

from collections.abc import Iterable

def flatten(xs):
    for x in xs:
        if isinstance(x, Iterable) and not isinstance(x, (str, bytes)):
            yield from flatten(x)
        else:
            yield x

此版本避免了 python 的递归限制（因此适用于任意深度的嵌套可迭代对象）。它是一个可以处理字符串和任意可迭代对象（甚至是无限可迭代对象）的生成器。flatten

import itertools
import collections

def flatten(iterable, ltypes=collections.Iterable):
    remainder = iter(iterable)
    while True:
        try:
            first = next(remainder)
        except StopIteration:
            break
        if isinstance(first, ltypes) and not isinstance(first, (str, bytes)):
            remainder = itertools.chain(first, remainder)
        else:
            yield first

以下是一些示例来演示其用法：

print(list(itertools.islice(flatten(itertools.repeat(1)),10)))
# [1, 1, 1, 1, 1, 1, 1, 1, 1, 1]

print(list(itertools.islice(flatten(itertools.chain(itertools.repeat(2,3),
                                       {10,20,30},
                                       'foo bar'.split(),
                                       itertools.repeat(1),)),10)))
# [2, 2, 2, 10, 20, 30, 'foo', 'bar', 1, 1]

print(list(flatten([[1,2,[3,4]]])))
# [1, 2, 3, 4]

seq = ([[chr(i),chr(i-32)] for i in range(ord('a'), ord('z')+1)] + list(range(0,9)))
print(list(flatten(seq)))
# ['a', 'A', 'b', 'B', 'c', 'C', 'd', 'D', 'e', 'E', 'f', 'F', 'g', 'G', 'h', 'H',
# 'i', 'I', 'j', 'J', 'k', 'K', 'l', 'L', 'm', 'M', 'n', 'N', 'o', 'O', 'p', 'P',
# 'q', 'Q', 'r', 'R', 's', 'S', 't', 'T', 'u', 'U', 'v', 'V', 'w', 'W', 'x', 'X',
# 'y', 'Y', 'z', 'Z', 0, 1, 2, 3, 4, 5, 6, 7, 8]

虽然可以处理无限生成器，但它不能处理无限嵌套：flatten

def infinitely_nested():
    while True:
        yield itertools.chain(infinitely_nested(), itertools.repeat(1))

print(list(itertools.islice(flatten(infinitely_nested()), 10)))
# hangs

@unutbu 非递归解决方案的生成器版本，正如 @Andrew 在评论中要求的那样：

def genflat(l, ltypes=collections.Sequence):
    l = list(l)
    i = 0
    while i < len(l):
        while isinstance(l[i], ltypes):
            if not l[i]:
                l.pop(i)
                i -= 1
                break
            else:
                l[i:i + 1] = l[i]
        yield l[i]
        i += 1

此生成器的略微简化版本：

def genflat(l, ltypes=collections.Sequence):
    l = list(l)
    while l:
        while l and isinstance(l[0], ltypes):
            l[0:1] = l[0]
        if l: yield l.pop(0)

def flatten(xs):
    res = []
    def loop(ys):
        for i in ys:
            if isinstance(i, list):
                loop(i)
            else:
                res.append(i)
    loop(xs)
    return res

我更喜欢简单的答案。没有发电机。没有递归或递归限制。只是迭代：

def flatten(TheList):
    listIsNested = True

    while listIsNested:                 #outer loop
        keepChecking = False
        Temp = []

        for element in TheList:         #inner loop
            if isinstance(element,list):
                Temp.extend(element)
                keepChecking = True
            else:
                Temp.append(element)

        listIsNested = keepChecking     #determine if outer loop exits
        TheList = Temp[:]

    return TheList

这适用于两个列表：内部 for 循环和外部 while 循环。

内部 for 循环遍历列表。如果它找到一个列表元素，它 （1） 使用 list.extend（） 将该部分扁平化一级嵌套，（2） 将 keepChecking 切换为 True。keepchecking 用于控制外部 while 循环。如果外部循环设置为 true，则会触发内部循环进行另一次传递。

这些传递会一直发生，直到找不到更多的嵌套列表。当最终在未找到任何通道的情况下发生传递时，keepChecking 永远不会跳闸为 true，这意味着 listIsNested 保持 false 并且外部 while 循环退出。

然后返回扁平化列表。

试运行

flatten([1,2,3,4,[100,200,300,[1000,2000,3000]]])

[1, 2, 3, 4, 100, 200, 300, 1000, 2000, 3000]

这是另一个更有趣的答案......

import re

def Flatten(TheList):
    a = str(TheList)
    b,_Anon = re.subn(r'[\[,\]]', ' ', a)
    c = b.split()
    d = [int(x) for x in c]

    return(d)

基本上，它将嵌套列表转换为字符串，使用正则表达式去除嵌套语法，然后将结果转换回（扁平化）列表。

虽然已经选择了一个优雅且非常蟒蛇般的答案，但我会提出我的解决方案，只是为了复习：

def flat(l):
    ret = []
    for i in l:
        if isinstance(i, list) or isinstance(i, tuple):
            ret.extend(flat(i))
        else:
            ret.append(i)
    return ret

请说出这段代码的好坏？

这是我的递归扁平化函数版本，它处理元组和列表，并允许您抛出任何位置参数的组合。返回一个生成器，该生成器按顺序生成整个序列，逐个 arg：

flatten = lambda *n: (e for a in n
    for e in (flatten(*a) if isinstance(a, (tuple, list)) else (a,)))

用法：

l1 = ['a', ['b', ('c', 'd')]]
l2 = [0, 1, (2, 3), [[4, 5, (6, 7, (8,), [9]), 10]], (11,)]
print list(flatten(l1, -2, -1, l2))
['a', 'b', 'c', 'd', -2, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]

我在这里没有看到这样的东西，只是从一个关于同一主题的封闭式问题中得出的，但为什么不做这样的事情（如果你知道你要拆分的列表类型）：

>>> a = [1, 2, 3, 5, 10, [1, 25, 11, [1, 0]]]    
>>> g = str(a).replace('[', '').replace(']', '')    
>>> b = [int(x) for x in g.split(',') if x.strip()]

你需要知道元素的类型，但我认为这可以概括，就速度而言，我认为它会更快。

使用递归和鸭子类型的生成器（针对 Python 3 进行了更新）：

def flatten(L):
    for item in L:
        try:
            yield from flatten(item)
        except TypeError:
            yield item

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

尝试创建一个可以在 Python 中扁平化不规则列表的函数很有趣，但当然这就是 Python 的用途（让编程变得有趣）。以下生成器运行良好，但有一些注意事项：

def flatten(iterable):
    try:
        for item in iterable:
            yield from flatten(item)
    except TypeError:
        yield iterable

它将展平您可能希望保留的数据类型（如 、 和 objects）。此外，该代码依赖于这样一个事实，即从不可迭代对象请求迭代器会引发 .bytearraybytesstrTypeError

>>> L = [[[1, 2, 3], [4, 5]], 6]
>>> def flatten(iterable):
    try:
        for item in iterable:
            yield from flatten(item)
    except TypeError:
        yield iterable


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

---

编辑：

我不同意以前的实现。问题在于，你不应该能够展平一些不可迭代的东西。这是令人困惑的，并给人以错误的论点印象。

>>> list(flatten(123))
[123]
>>>

下面的生成器与第一个生成器几乎相同，但没有尝试展平不可迭代对象的问题。正如人们所期望的那样，当给出不适当的参数时，它会失败。

def flatten(iterable):
    for item in iterable:
        try:
            yield from flatten(item)
        except TypeError:
            yield item

测试生成器可以正常使用提供的列表。但是，当新代码被赋予一个不可迭代的对象时，它将引发一个。下面显示了新行为的示例。TypeError

>>> L = [[[1, 2, 3], [4, 5]], 6]
>>> list(flatten(L))
[1, 2, 3, 4, 5, 6]
>>> list(flatten(123))
Traceback (most recent call last):
  File "<pyshell#32>", line 1, in <module>
    list(flatten(123))
  File "<pyshell#27>", line 2, in flatten
    for item in iterable:
TypeError: 'int' object is not iterable
>>>

以下是 2.7.5 中的实现：compiler.ast.flatten

def flatten(seq):
    l = []
    for elt in seq:
        t = type(elt)
        if t is tuple or t is list:
            for elt2 in flatten(elt):
                l.append(elt2)
        else:
            l.append(elt)
    return l

有更好、更快的方法（如果你已经到达这里，你已经看到了它们）

另请注意：

自 2.6 版起不推荐使用：编译器包已在 Python 3 中删除。

下面是一个简单的函数，可以展平任意深度的列表。无递归，避免堆栈溢出。

from copy import deepcopy

def flatten_list(nested_list):
    """Flatten an arbitrarily nested list, without recursion (to avoid
    stack overflows). Returns a new list, the original list is unchanged.

    >> list(flatten_list([1, 2, 3, [4], [], [[[[[[[[[5]]]]]]]]]]))
    [1, 2, 3, 4, 5]
    >> list(flatten_list([[1, 2], 3]))
    [1, 2, 3]

    """
    nested_list = deepcopy(nested_list)

    while nested_list:
        sublist = nested_list.pop(0)

        if isinstance(sublist, list):
            nested_list = sublist + nested_list
        else:
            yield sublist

完全是黑客，但我认为它会起作用（取决于你的data_type）

flat_list = ast.literal_eval("[%s]"%re.sub("[\[\]]","",str(the_list)))

这是另一种 py2 方法，我不确定它是最快的还是最优雅的，也不是最安全的......

from collections import Iterable
from itertools import imap, repeat, chain


def flat(seqs, ignore=(int, long, float, basestring)):
    return repeat(seqs, 1) if any(imap(isinstance, repeat(seqs), ignore)) or not isinstance(seqs, Iterable) else chain.from_iterable(imap(flat, seqs))

它可以忽略您想要的任何特定（或派生）类型，它返回一个迭代器，因此您可以将其转换为任何特定的容器，例如列表、元组、字典或简单地使用它以减少内存占用，无论好坏，它都可以处理初始不可迭代对象，例如 int ...

请注意，大部分繁重的工作都是在 C 中完成的，因为据我所知，这就是 itertools 的实现方式，所以虽然它是递归的，但 AFAIK 不受 python 递归深度的限制，因为函数调用发生在 C 中，尽管这并不意味着您受内存限制，尤其是在 OS X 中，其堆栈大小截至今天有硬性限制（OS X Mavericks）......

有一种稍微快一点的方法，但可移植性较差，只有在您可以假设可以显式确定输入的基本元素时才使用它，否则，您将获得无限递归，而堆栈大小有限的 OS X 将很快抛出分段错误......

def flat(seqs, ignore={int, long, float, str, unicode}):
    return repeat(seqs, 1) if type(seqs) in ignore or not isinstance(seqs, Iterable) else chain.from_iterable(imap(flat, seqs))

在这里，我们使用集合来检查类型，因此需要 O（1） vs O（类型数）来检查是否应该忽略一个元素，当然，任何具有所述忽略类型的派生类型的值都会失败，这就是它使用的原因，所以请谨慎使用它......strunicode

测试：

import random

def test_flat(test_size=2000):
    def increase_depth(value, depth=1):
        for func in xrange(depth):
            value = repeat(value, 1)
        return value

    def random_sub_chaining(nested_values):
        for values in nested_values:
            yield chain((values,), chain.from_iterable(imap(next, repeat(nested_values, random.randint(1, 10)))))

    expected_values = zip(xrange(test_size), imap(str, xrange(test_size)))
    nested_values = random_sub_chaining((increase_depth(value, depth) for depth, value in enumerate(expected_values)))
    assert not any(imap(cmp, chain.from_iterable(expected_values), flat(chain(((),), nested_values, ((),)))))

>>> test_flat()
>>> list(flat([[[1, 2, 3], [4, 5]], 6]))
[1, 2, 3, 4, 5, 6]
>>>  

$ uname -a
Darwin Samys-MacBook-Pro.local 13.3.0 Darwin Kernel Version 13.3.0: Tue Jun  3 21:27:35 PDT 2014; root:xnu-2422.110.17~1/RELEASE_X86_64 x86_64
$ python --version
Python 2.7.5

我使用递归来解决任何深度的嵌套列表

def combine_nlist(nlist,init=0,combiner=lambda x,y: x+y):
    '''
    apply function: combiner to a nested list element by element(treated as flatten list)
    '''
    current_value=init
    for each_item in nlist:
        if isinstance(each_item,list):
            current_value =combine_nlist(each_item,current_value,combiner)
        else:
            current_value = combiner(current_value,each_item)
    return current_value

所以在我定义函数combine_nlist后，很容易使用这个函数做扁平化。或者您可以将其组合成一个函数。我喜欢我的解决方案，因为它可以应用于任何嵌套列表。

def flatten_nlist(nlist):
    return combine_nlist(nlist,[],lambda x,y:x+[y])

结果

In [379]: flatten_nlist([1,2,3,[4,5],[6],[[[7],8],9],10])
Out[379]: [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

在不使用任何库的情况下：

def flat(l):
    def _flat(l, r):    
        if type(l) is not list:
            r.append(l)
        else:
            for i in l:
                r = r + flat(i)
        return r
    return _flat(l, [])



# example
test = [[1], [[2]], [3], [['a','b','c'] , [['z','x','y']], ['d','f','g']], 4]    
print flat(test) # prints [1, 2, 3, 'a', 'b', 'c', 'z', 'x', 'y', 'd', 'f', 'g', 4]

用：itertools.chain

import itertools
from collections import Iterable

def list_flatten(lst):
    flat_lst = []
    for item in itertools.chain(lst):
        if isinstance(item, Iterable):
            item = list_flatten(item)
            flat_lst.extend(item)
        else:
            flat_lst.append(item)
    return flat_lst

或者不带链接：

def flatten(q, final):
    if not q:
        return
    if isinstance(q, list):
        if not isinstance(q[0], list):
            final.append(q[0])
        else:
            flatten(q[0], final)
        flatten(q[1:], final)
    else:
        final.append(q)

我很惊讶没有人想到这一点。该死的递归，我没有得到这里的高级人员做出的递归答案。无论如何，这是我对此的尝试。需要注意的是，它非常特定于 OP 的用例

import re

L = [[[1, 2, 3], [4, 5]], 6]
flattened_list = re.sub("[\[\]]", "", str(L)).replace(" ", "").split(",")
new_list = list(map(int, flattened_list))
print(new_list)

输出：

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

最简单的方法是使用 morph 库。pip install morph

代码为：

import morph

list = [[[1, 2, 3], [4, 5]], 6]
flattened_list = morph.flatten(list)  # returns [1, 2, 3, 4, 5, 6]

我没有在这里浏览所有已经可用的答案，但这里有一个我想出的行，借用了 lisp 的第一个和其余列表处理方式

def flatten(l): return flatten(l[0]) + (flatten(l[1:]) if len(l) > 1 else []) if type(l) is list else [l]

这是一个简单和一个不那么简单的案例——

>>> flatten([1,[2,3],4])
[1, 2, 3, 4]

>>> flatten([1, [2, 3], 4, [5, [6, {'name': 'some_name', 'age':30}, 7]], [8, 9, [10, [11, [12, [13, {'some', 'set'}, 14, [15, 'some_string'], 16], 17, 18], 19], 20], 21, 22, [23, 24], 25], 26, 27, 28, 29, 30])
[1, 2, 3, 4, 5, 6, {'age': 30, 'name': 'some_name'}, 7, 8, 9, 10, 11, 12, 13, set(['set', 'some']), 14, 15, 'some_string', 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30]
>>> 

我知道已经有很多很棒的答案，但我想添加一个使用函数式编程方法来解决问题的答案。在这个答案中，我使用了双递归：

def flatten_list(seq):
    if not seq:
        return []
    elif isinstance(seq[0],list):
        return (flatten_list(seq[0])+flatten_list(seq[1:]))
    else:
        return [seq[0]]+flatten_list(seq[1:])

print(flatten_list([1,2,[3,[4],5],[6,7]]))

输出：

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

我不确定这是否一定更快或更有效，但这就是我所做的：

def flatten(lst):
    return eval('[' + str(lst).replace('[', '').replace(']', '') + ']')

L = [[[1, 2, 3], [4, 5]], 6]
print(flatten(L))

这里的函数将列表变成一个字符串，去掉所有的方括号，将方括号附加回两端，然后把它转回一个列表。flatten

虽然，如果你知道你的列表中会有方括号的字符串，比如 ，你将不得不做其他事情。[[1, 2], "[3, 4] and [5]"]

您可以使用第三方软件包中的 deepflatten iteration_utilities：

>>> from iteration_utilities import deepflatten
>>> L = [[[1, 2, 3], [4, 5]], 6]
>>> list(deepflatten(L))
[1, 2, 3, 4, 5, 6]

>>> list(deepflatten(L, types=list))  # only flatten "inner" lists
[1, 2, 3, 4, 5, 6]

它是一个迭代器，因此您需要迭代它（例如，通过将其包装或在循环中使用它）。在内部，它使用迭代方法而不是递归方法，并且它被编写为 C 扩展，因此它可以比纯 python 方法更快：list

>>> %timeit list(deepflatten(L))
12.6 µs ± 298 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)
>>> %timeit list(deepflatten(L, types=list))
8.7 µs ± 139 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

>>> %timeit list(flatten(L))   # Cristian - Python 3.x approach from https://stackoverflow.com/a/2158532/5393381
86.4 µs ± 4.42 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

>>> %timeit list(flatten(L))   # Josh Lee - https://stackoverflow.com/a/2158522/5393381
107 µs ± 2.99 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

>>> %timeit list(genflat(L, list))  # Alex Martelli - https://stackoverflow.com/a/2159079/5393381
23.1 µs ± 710 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)

---

我是该库的作者。iteration_utilities

在尝试回答这样的问题时，您确实需要给出您提出的代码作为解决方案的局限性。如果只是关于性能，我不会太介意，但大多数作为解决方案提出的代码（包括公认的答案）都无法展平任何深度大于 1000 的列表。

当我说大多数代码时，我指的是所有使用任何形式的递归（或调用递归的标准库函数）的代码。所有这些代码都失败了，因为对于每个递归调用，（调用）堆栈都会增长一个单位，并且（默认）python 调用堆栈的大小为 1000。

如果您不太熟悉调用堆栈，那么以下内容可能会有所帮助（否则您可以滚动到实现）。

调用堆栈大小和递归编程（地下城类比）

寻找宝藏并退出

想象一下，你进入一个巨大的地牢，里面有编号的房间，寻找宝藏。你不知道这个地方，但你有一些关于如何找到宝藏的迹象。每个指示都是一个谜语（难度各不相同，但你无法预测它们会有多难）。你决定考虑一下节省时间的策略，你做了两个观察：

1. 找到宝藏是很困难的（漫长的），因为你必须解决（可能很难的）谜语才能到达那里。
2. 一旦找到宝藏，回到入口可能很容易，你只需要在另一个方向使用相同的路径（尽管这需要一点记忆来回忆你的路径）。

进入地牢时，你会注意到这里有一个小笔记本。你决定用它来写下你在解开谜语后离开的每个房间（进入一个新房间时），这样你就可以回到入口处。这是一个天才的想法，你甚至不会花一分钱来实施你的策略。

你进入地牢，成功地解决了前 1001 个谜语，但这里有一些你没有计划的东西，你借来的笔记本里没有空间了。你决定放弃你的任务，因为你宁愿没有宝藏，也不愿永远迷失在地牢里（这看起来确实很聪明）。

执行递归程序

基本上，这与寻找宝藏完全相同。地牢是计算机的内存，你现在的目标不是找到宝藏，而是计算一些函数（为给定的 x 找到 f（x）。指示只是可以帮助您求解 f（x） 的子例程。您的策略与调用堆栈策略相同，笔记本是堆栈，房间是函数的返回地址：

x = ["over here", "am", "I"]
y = sorted(x) # You're about to enter a room named `sorted`, note down the current room address here so you can return back: 0x4004f4 (that room address looks weird)
# Seems like you went back from your quest using the return address 0x4004f4
# Let's see what you've collected 
print(' '.join(y))

你在地牢中遇到的问题在这里是一样的，调用堆栈的大小是有限的（这里是 1000），因此，如果你输入了太多函数而没有返回，那么你将填满调用堆栈并出现一个错误，看起来像“亲爱的冒险家，我很抱歉，但你的笔记本已满”：。请注意，您不需要递归来填充调用堆栈，但非递归程序调用 1000 个函数而不返回的可能性很小。同样重要的是要了解，一旦您从函数返回，调用堆栈就会从使用的地址中释放出来（因此得名“堆栈”，返回地址在进入函数之前被推入，在返回时被拉出）。在简单递归的特殊情况下（一个函数，一遍又一遍地调用自己），你将一遍又一遍地输入，直到计算完成（直到找到宝藏），然后从返回，直到你回到你最初调用的地方。调用堆栈永远不会从任何内容中释放出来，直到最后它将一个接一个地从所有返回地址中释放出来。RecursionError: maximum recursion depth exceededffff

如何避免这个问题？

这其实很简单：“如果你不知道递归能有多深，就不要使用递归”。这并不总是正确的，因为在某些情况下，尾调用递归可以优化 （TCO）。但在 python 中，情况并非如此，即使是“写得很好”的递归函数也不会优化堆栈的使用。关于这个问题，Guido 发表了一篇有趣的文章：Tail Recursion Elimination。

有一种技术可以使任何递归函数迭代，这种技术我们可以称之为自带笔记本。例如，在我们的特定情况下，我们只是在探索一个列表，进入一个房间相当于进入一个子列表，你应该问自己的问题是，我怎样才能从一个列表回到它的父列表？答案并不复杂，重复以下操作，直到 为空：stack

1. 在输入新的子列表时推送当前列表和 in（注意列表地址+索引也是一个地址，因此我们只是使用与调用堆栈完全相同的技术）;addressindexstack
2. 每次找到项目时，它（或将它们添加到列表中）;yield
3. 完全浏览列表后，使用返回地址（和索引）返回到父列表。stack

另请注意，这相当于树中的 DFS，其中某些节点是子列表，而某些节点是简单项：（for ）。树如下所示：A = [1, 2]0, 1, 2, 3, 4L = [0, [1,2], 3, 4]

                    L
                    |
           -------------------
           |     |     |     |
           0   --A--   3     4
               |   |
               1   2

DFS 遍历预购顺序为：L、0、A、1、2、3、4。 请记住，为了实现迭代 DFS，您还需要一个堆栈。我之前提出的实现导致具有以下状态（对于 和 ）：stackflat_list

init.:  stack=[(L, 0)]
**0**:  stack=[(L, 0)],         flat_list=[0]
**A**:  stack=[(L, 1), (A, 0)], flat_list=[0]
**1**:  stack=[(L, 1), (A, 0)], flat_list=[0, 1]
**2**:  stack=[(L, 1), (A, 1)], flat_list=[0, 1, 2]
**3**:  stack=[(L, 2)],         flat_list=[0, 1, 2, 3]
**3**:  stack=[(L, 3)],         flat_list=[0, 1, 2, 3, 4]
return: stack=[],               flat_list=[0, 1, 2, 3, 4]

在此示例中，堆栈最大大小为 2，因为输入列表（以及树）的深度为 2。

实现

对于实现，在 python 中，您可以通过使用迭代器而不是简单列表来简化一点。对（子）迭代器的引用将用于存储子列表返回地址（而不是同时具有列表地址和索引）。这不是一个很大的区别，但我觉得这更具可读性（而且速度也更快）：

def flatten(iterable):
    return list(items_from(iterable))

def items_from(iterable):
    cursor_stack = [iter(iterable)]
    while cursor_stack:
        sub_iterable = cursor_stack[-1]
        try:
            item = next(sub_iterable)
        except StopIteration:   # post-order
            cursor_stack.pop()
            continue
        if is_list_like(item):  # pre-order
            cursor_stack.append(iter(item))
        elif item is not None:
            yield item          # in-order

def is_list_like(item):
    return isinstance(item, list)

另外，请注意，在我有 ，可以更改它以处理更多输入类型，这里我只想拥有最简单的版本，其中（可迭代）只是一个列表。但你也可以这样做：is_list_likeisinstance(item, list)

def is_list_like(item):
    try:
        iter(item)
        return not isinstance(item, str)  # strings are not lists (hmm...) 
    except TypeError:
        return False

这会将字符串视为“简单项”，因此将返回而不是 .请注意，在这种情况下，每个项目都会被调用两次，让我们假设这是读者的练习，以使其更干净。flatten_iter([["test", "a"], "b])["test", "a", "b"]["t", "e", "s", "t", "a", "b"]iter(item)

对其他实现的测试和注释

最后，请记住，您不能使用 打印无限嵌套的列表，因为它在内部会使用递归调用 （）。出于同样的原因，涉及的解决方案将失败，并显示相同的错误消息。Lprint(L)__repr__RecursionError: maximum recursion depth exceeded while getting the repr of an objectflattenstr

如果需要测试解决方案，可以使用此函数生成一个简单的嵌套列表：

def build_deep_list(depth):
    """Returns a list of the form $l_{depth} = [depth-1, l_{depth-1}]$
    with $depth > 1$ and $l_0 = [0]$.
    """
    sub_list = [0]
    for d in range(1, depth):
        sub_list = [d, sub_list]
    return sub_list

这给了：>>> .build_deep_list(5)[4, [3, [2, [1, [0]]]]]

没有递归或嵌套循环。几行。格式良好且易于阅读：

def flatten_deep(arr: list):
    """ Flattens arbitrarily-nested list `arr` into single-dimensional. """

    while arr:
        if isinstance(arr[0], list):  # Checks whether first element is a list
            arr = arr[0] + arr[1:]  # If so, flattens that first element one level
        else:
            yield arr.pop(0)  # Otherwise yield as part of the flat array

flatten_deep(L)

来自我自己在 https://github.com/jorgeorpinel/flatten_nested_lists/blob/master/flatten.py 的代码

只需使用一个有趣的库：pip install funcy

import funcy


funcy.flatten([[[[1, 1], 1], 2], 3]) # returns generator
funcy.lflatten([[[[1, 1], 1], 2], 3]) # returns list

Pandas 有一个函数可以做到这一点。如您所提到的，它返回一个迭代器。

In [1]: import pandas
In [2]: pandas.core.common.flatten([[[1, 2, 3], [4, 5]], 6])
Out[2]: <generator object flatten at 0x7f12ade66200>
In [3]: list(pandas.core.common.flatten([[[1, 2, 3], [4, 5]], 6]))
Out[3]: [1, 2, 3, 4, 5, 6]

我经常使用 more_itertools.collapse：

# pip install more-itertools
from more_itertools import collapse

out = list(collapse([[[1, 2, 3], [4, 5]], 6]))
# [1, 2, 3, 4, 5, 6]

它处理开箱即用的字符串/字节，并且不会将它们解压缩为单个字符：

list(collapse([[[1, 2, 3, 'abc'], [4, 5]], 6, 'def']))
# [1, 2, 3, 'abc', 4, 5, 6, 'def']

它也可以停止在给定的嵌套级别：

list(collapse([[[1, 2, 3], [4, 5]], 6], levels=1))
# [[1, 2, 3], [4, 5], 6]

完整的代码相对简单，并且还使用了递归方法：

def collapse(iterable, base_type=None, levels=None):
    def walk(node, level):
        if (
            ((levels is not None) and (level > levels))
            or isinstance(node, (str, bytes))
            or ((base_type is not None) and isinstance(node, base_type))
        ):
            yield node
            return

        try:
            tree = iter(node)
        except TypeError:
            yield node
            return
        else:
            for child in tree:
                yield from walk(child, level + 1)

    yield from walk(iterable, 0)