如何找到第五列最大值的索引？

Question

我的数组 A 为：

    A = [[0.0, 1.0, 3.0, -1.0, -1008.0],
[0.0, 1.0, 3.0, -1.0, -1008.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, 0.0, -1168.0],
[0.0, 1.0, 3.0, -2.0, -1488.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, 0.0, -1168.0],
[0.0, 1.0, 3.0, -4.0, -1808.0],
[0.0, 1.0, 3.0, -2.0, -1488.0],
[0.0, 1.0, 3.0, -3.0, -1648.0],
[0.0, 1.0, 3.0, -3.0, -1648.0],
[0.0, 1.0, 3.0, -3.0, -1648.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, 3.0, -1648.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, 2.0, -1488.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, -2.0, -1488.0],
[0.0, 1.0, 3.0, 2.0, -1488.0],
[0.0, 1.0, 3.0, 3.0, -1648.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, 2.0, -1488.0],
[0.0, 1.0, 3.0, 0.0, -1168.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, -3.0, -1648.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, -1.0, -1328.0],
[0.0, 1.0, 3.0, -2.0, -1488.0],
[0.0, 1.0, 3.0, 0.0, -1168.0],
[0.0, 1.0, 3.0, -2.0, -1488.0],
[0.0, 1.0, 3.0, -3.0, -1648.0],
[0.0, 1.0, 3.0, -2.0, -1488.0],
[0.0, 1.0, 3.0, 0.0, -1168.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, 0.0, -1168.0],
[0.0, 1.0, 3.0, 1.0, -1328.0],
[0.0, 1.0, 3.0, 0.0, -1168.0],
[0.0, 1.0, 3.0, 0.0, -1168.0],
[0.0, 1.0, 3.0, 0.0, -1168.0]]

我想找到第 4 列（第五列）中的最大值出现的位置，并且我想最终得到这个结果，

m = [0 , 1]

这是发生两个-1008.0 的地方。我尝试过使用：

m = np.argwhere(A == np.amax(A))

但我似乎没有正确使用它。有人可以帮我吗？

Answer 1

我会这样做：

last_col = np.array(A)[:, -1] # take every row in `A`, but only the last column
print(np.argwhere(last_col == np.amax(last_col)).flatten().tolist())