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')

Answer 2

如果您的意思是搜索“c”：

numpy.where(ar == 'c')

Answer 3

ar = [['0', '1', '2', '3'],
 ['1', 'a', 'b', 'b'],
 ['2', 'b', 'c', 'd']]

print(ar[2][2])

以上仅用于访问第 2 列第 2 行中的“c”，如果您想访问整个列，则必须：

for i in range(0, 3):
    print(ar[2][i])