2012-06-20 21 views
6

Mam macierz 800x800 i chcę analizować tylko te elementy w jej części zewnętrznej. Potrzebuję nowej tablicy bez elementów plasterka [5: -5, 5: -5]. Nie musi koniecznie zwracać tablicy 2d, a także płaskiej tablicy lub listy. Przykład:Jak usunąć podzbiór tablicy 2d?

import numpy 

>>> a = numpy.arange(1,10) 
array([1, 2, 3, 4, 5, 6, 7, 8, 9]) 

>>> a.shape = (3,3) 
array([[1, 2, 3], 
    [4, 5, 6], 
    [7, 8, 9]]) 

muszę odrzucić podstawowe elementy, coś jak:

del a[1:2,1:2] 

spodziewam się mieć:

array([1, 2, 3, 4, 6, 7, 8, 9]) 

Próbowałem użyć numpy.delete(), ale wydaje się działać dla jednej osi na raz. Zastanawiam się, czy istnieje bardziej prosty sposób na zrobienie tego.

Odpowiedz

2

Można wymienić środkowy obszar o pewnej wartości zastępczego (użyłem -12345, że nic nie może występować w swoich rzeczywistych danych będzie działać), a następnie wybrać wszystko, co nie jest równa tej wartości:

>>> import numpy as np 
>>> a = np.arange(1,26) 
>>> a.shape = (5,5) 
>>> a 
array([[ 1, 2, 3, 4, 5], 
     [ 6, 7, 8, 9, 10], 
     [11, 12, 13, 14, 15], 
     [16, 17, 18, 19, 20], 
     [21, 22, 23, 24, 25]]) 

>>> a[1:4,1:4] = -12345 
>>> a 
array([[  1,  2,  3,  4,  5], 
     [  6, -12345, -12345, -12345,  10], 
     [ 11, -12345, -12345, -12345,  15], 
     [ 16, -12345, -12345, -12345,  20], 
     [ 21,  22,  23,  24,  25]]) 
>>> a[a != -12345] 
array([ 1, 2, 3, 4, 5, 6, 10, 11, 15, 16, 20, 21, 22, 23, 24, 25]) 

Jeśli używasz tablicę pływaka zamiast tablicy liczb całkowitych, można to zrobić trochę bardziej elegancko przy użyciu NaN i isfinite:

>>> a = np.arange(1,26).astype('float32') 
>>> a.shape = (5,5) 
>>> a[1:4,1:4] = np.nan 
>>> a 
array([[ 1., 2., 3., 4., 5.], 
     [ 6., nan, nan, nan, 10.], 
     [ 11., nan, nan, nan, 15.], 
     [ 16., nan, nan, nan, 20.], 
     [ 21., 22., 23., 24., 25.]], dtype=float32) 
>>> a[np.isfinite(a)] 
array([ 1., 2., 3., 4., 5., 6., 10., 11., 15., 16., 20., 
    21., 22., 23., 24., 25.], dtype=float32) 
6

można użyć logiczną tablicę do indeksowania tablicy jakikolwiek sposób chcesz. W ten sposób nie musisz zmieniać żadnych wartości w oryginalnej tablicy, jeśli nie chcesz. Oto prosty przykład:

>>> import numpy as np 
>>> a = np.arange(1,10).reshape(3,3) 
>>> b = a.astype(bool) 
>>> b[1:2,1:2] = False 
>>> b 
array([[ True, True, True], 
     [ True, False, True], 
     [ True, True, True]], dtype=bool) 
>>> a[b] 
array([1, 2, 3, 4, 6, 7, 8, 9]) 
+0

wielki wybór, bo nie trzeba modyfikować de oryginalną tablicę i nie ma potrzeby znać zawartość tablicy – user1470350

+0

@ user1470350 - Tak, jest to dobry sposób, aby robić rzeczy :) – fraxel