如何找到 numpy 数组中行索引的最大值?

How do I find the maximum value of a row index in a numpy array?

我有以下数组:

a = [[0.         0.        ]
     [0.22333333 0.14333333]
     [0.48166667 0.99      ]
     [0.735      1.66666667]
     [1.02666667 2.16666667]
     [1.43833333 2.66333333]
     [1.71666667 3.15666667]]

我希望能够找到特定行索引处的最大值。例如,第 2 行的最大值为 0.99,第 6 行的最大值为 3.15。

我知道我可以使用

np.max(array,axis=1)

产生这个:

[0.         0.22333333 0.99       1.66666667 2.16666667 2.66333333
  3.15666667]

但我只想要我感兴趣的行的一个值。例如,如果我只想要数组中第 2 行的最大值,那么输出值应该如下所示:

0.99

那么只需在您想要的那一行应用 max 即可?例如array[2].max()

或类似的东西

np.amax(一个[行])