Mam sytuację, w której mam dane, które czasami mogą być zagnieżdżone w wielu warstwach macierzy.Najprostszy sposób na zwrócenie tablicy zagnieżdżonej w wielu tablicach
Kilka razy dane mogą być zagnieżdżone jak:
[ [ 'green', 'blue', 'red' ] ]
Inne czasy
[[[ ['green', 'blue', 'red' ] ]]]
chcę wyodrębnić tablicy i zwraca go, co byłoby najbardziej pythonic sposób to zrobić?
Numpy jest twoim najlepszym przyjacielem, jak zawsze:
>>> import numpy as np
>>> a = [[[ ['green', 'blue', 'red' ] ]]]
>>> print np.squeeze(a)
['green' 'blue' 'red']
Funkcja numpy squeeze() usunąć wszystkie wymiary, które są 1 w twojej tablicy.
Nie zgadzam się z "twoim najlepszym przyjacielem jak zawsze" - ale +1 –
:) dlaczego? Numpy kiedykolwiek zawiodła cię w przeszłości? –
def get_nested_list(a):
if len(a) == 1 and isinstance(a[0], list):
return get_nested_list(a[0])
return a
Przykłady:
>>> get_nested_list([[[ ['green', 'blue', 'red' ] ]]])
['green', 'blue', 'red']
>>> get_nested_list([[[[1, 2],[3]]]])
[[1, 2], [3]]
Jest to użyteczne tylko w przypadku zagnieżdżenia pojedynczego elementu. – asheeshr
@AshRj: Teraz obsługuje listy wieloelementowe z pewnym zdrowiem. –
Zakładam, że pierwszym testem jest sprawdzenie pustych list? Ale istnieje już kontrola len (a) == 1, która obsłuży puste listy .. – asheeshr
rekurencyjne rozwiązanie:
def strip(a):
if len(a)==1:
if isinstance(a[0], list):
a=a[0]
return strip(a)
else:
return a
return a
Czy są jakieś inne elementy na listach nadrzędnych? czy po prostu chcesz pozbyć się nadmiernego zagnieżdżenia? –
Zobacz http://stackoverflow.com/questions/952914/making-a-flat-list-out-of-list-in-python –
Pozbycie się "zagnieżdżania" jest łatwe - ale wyglądałbym bardziej nie mieć go na pierwszym miejscu, tj. jeśli nieprzewidywalne wyniki - wtedy jest jakaś luka, która wymaga adresowania ... –