使用行的第一个元素查找行号

Question

我有一个numpy矩阵，我将某种键存储在每行的第一个元素中（或者以其他方式将所有键存储在第一列中）。

[[123,0,1,1,2],
[12,1,2,3,4],
[1,0,2,5,4],
[90,1,1,4,3]]

我想通过密钥获取行号。我发现我们可以使用numpy.where但不清楚如何使用它来获取行号。我想要像

这样的东西

>>numpy.func(myMatrix,90)
3

任何帮助都将不胜感激。

Answer 1

根据online doc，setStatusBarGravity(Gravity.CENTER | 10) 如果你只给它一个布尔数组，将返回numpy.where元素的坐标列表（每个维度一个列表）。 因此，我们可以通过抓取数组的第一列，将其与要查找的元素进行比较并在该布尔数组上调用np.where来获取所需的信息。所有这些看起来都是这样的：

True

Answer 2

将第一列与90中的np.where进行比较。它将返回与90相等的项目索引数组：

In [3]: A = np.array([[123,0,1,1,2],
[12,1,2,3,4],
[1,0,2,5,4],
[90,1,1,4,3]])


In [6]: np.where(A[:,0]==90)[0]
Out[6]: array([3])