python矩阵搜索

Question

如果我有以下矩阵:

import numpy

ar = numpy.array((('0','1','2','3'), ('1','a','b','b'), ('2','b','c','d')), str)
print(ar)

输出:

[['0' '1' '2' '3']
 ['1' 'a' 'b' 'b']
 ['2' 'b' 'c' 'd']]

我想获取满足行和列条件的值。标题行是列(可以是字符串)，左边的第一列是所有行(可以是字符串)。因此，如果列为“2”，行为“2”，我将得到“c”。我不知道“c”，但只知道行和列的值。我该怎么做？

Answer 1

你可以使用numpy.where:

In [7]: numpy.where(ar == 'c')
Out[7]: (array([2]), array([2]))

和:

In [8]: ar[numpy.where(ar == 'c')]
Out[8]: 
array(['c'], 
      dtype='|S1')