numpy.ndarray.argmax

ndarray.argmax(axis=None, out=None)

沿给定轴的最大值的返回指数。

有关完整文档，请参阅numpy.argmax。

也可以看看

numpy.argmax

等效函数