python - 返回嵌套在多个数组中的数组的最简单方法

Question

我有一种情况，我的数据有时可以嵌套在多个数组层中。

有时数据可以嵌套如下:

[ [ 'green', 'blue', 'red' ] ]

其他时间

[[[ ['green', 'blue', 'red' ] ]]]

我想提取数组并返回它，最pythonic的方法是什么？

Answer 1

Numpy 一如既往地是你最好的 friend :

>>> import numpy as np
>>> a = [[[ ['green', 'blue', 'red' ] ]]]
>>> print np.squeeze(a)
['green' 'blue' 'red']

numpy 函数 squeeze()删除数组中所有为 1 的维度。