最簡單的方法來返回嵌套在多個數組中的數組

Question

我有一種情況，我有時可以嵌套在多個數組層中的數據。

有些時候，數據可以被嵌套，如：

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

其他時間

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

我想提取數組並返回，這將是這樣做的最Python的方式？

Answer 1

numpy的是一直是你最好的朋友：

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

的numpy的功能squeeze()刪除所有維度在你的數組中是1。

Answer 2

def get_nested_list(a): 
    if len(a) == 1 and isinstance(a[0], list): 
     return get_nested_list(a[0]) 
    return a 

例子：

>>> get_nested_list([[[ ['green', 'blue', 'red' ] ]]]) 
['green', 'blue', 'red'] 
>>> get_nested_list([[[[1, 2],[3]]]]) 
[[1, 2], [3]] 

Answer 3

遞歸解決方案：

def strip(a): 
    if len(a)==1: 
     if isinstance(a[0], list): 
      a=a[0] 
      return strip(a) 
     else: 
      return a 
    return a