在python中展平列表

时间:2012-10-15 09:48:22

标签: python list

我想要整理清单:

exampleArray = [[[151.68694121866872]], 
                [[101.59534468349297]], 
                [[72.16055999176308]]]

为:

[151.68694121866872, 101.59534468349297, 72.16055999176308]

现在我正在这样做:

resultArray= list(chain.from_iterable(list(chain.from_iterable(exampleArray))))

尽管它有效但我想知道是否有更好的方法。

5 个答案:

答案 0 :(得分:2)

怎么样

result = [x[0][0] for x in exampleArray]

答案 1 :(得分:1)

In [6]: from itertools import chain

In [7]: lis=[[[151.68694121866872]], [[101.59534468349297]], [[72.16055999176308]]]

In [8]: list(chain(*(chain(*lis))))

Out[8]: [151.68694121866872, 101.59534468349297, 72.16055999176308]

答案 2 :(得分:1)

没有更好的方法,这是最推荐的方式。见official recipes

def flatten(listOfLists):
    "Flatten one level of nesting"
    return chain.from_iterable(listOfLists)

答案 3 :(得分:1)

您无需将itertools.chain对象(可迭代)转换为列表:

resultArray= list(chain.from_iterable(list(chain.from_iterable(exampleArray))))
# could be rewritten as
resultArray= list(chain.from_iterable(chain.from_iterable(exampleArray)))

您可以使用递归编写深度函数:

def deep_chain_from_iterable(it, n):
    if n == 0:
        return list(it)
    else:
        return deep_chain_from_iterable(itertools.chain.from_iterable(it),n-1)

deep_chain_from_iterable(exampleArray, 2)

答案 4 :(得分:0)

对于固定的深度级别(如示例所示),您可以总结足够的时间:

sum(sum(listoflistsoflists, []), [])