python 查找最大值的坐标

可以使用NumPy库中的argmax函数来查找数组中最大值的索引，例如：

import numpy as np

a = np.array([1, 2, 3, 2, 1])
max_index = np.argmax(a)
print(max_index)  # 输出2，即数组a中最大值3的索引

如果是二维数组，可以指定查找的轴，例如：

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
max_index = np.argmax(a, axis=1)
print(max_index)  # 输出[2, 2]，即每行中最大值的索引