Numpy在2D矩阵上的where()

Question

我有一个像这样的矩阵

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

我想得到行包含字符串'bar'的索引

在1d数组中

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

应该给我指数[0,3]然后广播: -

results = t[rows]

应该给我正确的行

但我无法弄清楚如何让它与2d数组一起工作.

Answer 1

您必须将数组切片到要编制索引的列:

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

返回:

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

Answer 2

对于一般情况,您的搜索字符串可以位于任何列中,您可以执行以下操作:

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