Uzyskiwanie siatki macierzy poprzez indeksowanie logiczne w Numpy

Question

Próbuję przepisać funkcję za pomocą numpy, która jest oryginalnie w MATLAB. Jest to logiczne część indeksujący, który przedstawia się następująco w MATLAB:

X = reshape(1:16, 4, 4).'; 
idx = [true, false, false, true]; 
X(idx, idx) 

ans = 

    1  4 
    13 16 

Kiedy próbuję to zrobić w numpy, nie mogę uzyskać prawidłowe indeksowanie:

X = np.arange(1, 17).reshape(4, 4) 
idx = [True, False, False, True] 
X[idx, idx] 
# Output: array([6, 1, 1, 6]) 

Jaki jest właściwy sposób na uzyskanie siatka z macierzy poprzez indeksowanie logiczne?

Answer 1

Można również napisać:

>>> X[np.ix_(idx,idx)] 
array([[ 1, 4], 
     [13, 16]]) 

Answer 2

W numpy nazywa się to fancy indexing. Aby uzyskać pożądane elementy, należy użyć tablicy indeksów 2D.

Możesz użyć outer, aby utworzyć z własnej tablicy właściwą tablicę indeksów. The outers, po zastosowaniu do dwóch sekwencji 1D, porównaj każdy element jednej sekwencji z każdym elementem drugiego. Przypominając, że True*True=True i False*True=False Z np.multiply.outer(), która jest taka sama jak np.outer(), może dać Ci 2D indeksach:

idx_2D = np.outer(idx,idx) 
#array([[ True, False, False, True], 
#  [False, False, False, False], 
#  [False, False, False, False], 
#  [ True, False, False, True]], dtype=bool) 

które można wykorzystać:

x[ idx_2D ] 
array([ 1, 4, 13, 16]) 

w normalnym kodu można użyć x=[np.outer(idx,idx)] ale nie oszczędza pamięci, działa tak samo, jak gdyby po wykonaniu plasterka dodano del idx_2D.

Answer 3

In [1]: X = np.arange(1, 17).reshape(4, 4) 

In [2]: idx = np.array([True, False, False, True]) # note that here idx has to 
                # be an array (not a list) 
                # or boolean values will be 
                # interpreted as integers 

In [3]: X[idx][:,idx] 
Out[3]: 
array([[ 1, 4], 
     [13, 16]])