Numpy gdzie() na matrycy 2D

Question

mam matrycę jak ten

t = np.array([[1,2,3,'foo'], 
[2,3,4,'bar'], 
[5,6,7,'hello'], 
[8,9,1,'bar']]) 

chcę uzyskać indeksy gdzie wiersze zawierające ciąg „bar”

W 1d tablicy

rows = np.where(t == 'bar') 

powinny mi indeksy [0,3], a następnie nadawanie: -

results = t[rows] 

powinien dać mi właściwe wiersze

Ale nie mogę wymyślić, jak sprawić, by działał z tablicami 2d.

Answer 1

W ogólnym przypadku, gdy wyszukiwanie ciąg może być w dowolnej kolumnie, możesz to zrobić:

>>> rows, cols = np.where(t == 'bar') 
>>> t[rows] 
array([['2', '3', '4', 'bar'], 
     ['8', '9', '1', 'bar']], 
     dtype='|S11') 

Answer 2

Trzeba pokroić tablicę do kol chcesz index:

rows = np.where(t[:,3] == 'bar') 
result = t1[rows] 

Powrócisz:

[[2,3,4,'bar'], 
    [8,9,1,'bar']]