numpy Indexerror: zu viele Indizes für Array, wenn Indizieren Matrix mit einem anderen

Question

Ich habe eine Matrix A, die I wie folgt zu erstellen:

>>> a = np.matrix("1 2 3; 4 5 6; 7 8 9; 10 11 12") 

Ich habe eine Matrix Etiketten, die ich so erstellen:

>>> labels = np.matrix("1;0;1;1") 

Diese

ist das, was die beiden Matrizes wie folgt aussehen:

>>> a 
matrix([[ 1, 2, 3], 
     [ 4, 5, 6], 
     [ 7, 8, 9], 
     [10, 11, 12]]) 
>>> labels 
matrix([[1], 
     [0], 
     [1], 
     [1]]) 

wie Sie sehen können, wenn ich alle Spalten auswählen, gibt es keine p roblem

>>> a[labels == 1, :] 
matrix([[ 1, 7, 10]]) 

Aber wenn ich versuche, eine Spalte I ein Fehler

>>> a[labels == 1, 1] 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
    File "/usr/local/lib/python2.7/site-packages/numpy/matrixlib/defmatrix.py", line 305, in  __getitem__ 
    out = N.ndarray.__getitem__(self, index) 
IndexError: too many indices for array 
>>> 

Weiß jemand, warum dies bekommen angeben? Ich bin mir bewusst, dass es bereits ähnliche Fragen gibt, aber keine von ihnen erklärt mein Problem gut genug, noch sind die Antworten hilfreich für mich.

Answer 1

Da labels eine Matrix ist, wenn Sie labels==1 tun, erhalten Sie eine boolesche Matrix der gleichen Form. Dann a[labels==1, :] wird Ihnen nur die erste Spalte mit den Zeilen entsprechend der Übereinstimmung zurückgeben. Beachten Sie, dass Ihre Absicht zu erhalten:

matrix([[ 1, 2, 3], 
     [ 7, 8, 9], 
     [10, 11, 12]]) 

nicht erreicht wurde (nur die erste Spalte bekommen), obwohl es für NumPy arbeitete < 1.8 (wie von @seberg darauf hingewiesen).

Um zu bekommen, was Sie möchten, eine abgeflachte Ansicht labels verwenden:

a[labels.view(np.ndarray).ravel()==1, :]