我有一个这样的矩阵
t = np.array([[1,2,3,'foo'],
[2,3,4,'bar'],
[5,6,7,'hello'],
[8,9,1,'bar']])
我想获取行包含字符串“ bar”的索引
一维阵列
rows = np.where(t == 'bar')
应该给我指数[0,3],然后广播:
results = t[rows]
应该给我正确的行
但是我不知道如何使它与2d数组一起使用。
对于一般情况,您的搜索字符串可以位于任何列中,则可以执行以下操作:
>>> rows, cols = np.where(t == 'bar')
>>> t[rows]
array([['2', '3', '4', 'bar'],
['8', '9', '1', 'bar']],
dtype='|S11')
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句