Python小技巧 - argmax

argmax 返回的是输入列表中最大值的位置，其重要性不必多言，但是据我所知 Python 自带的库中只提供 max 这个函数，并没有 argmax，但是实现起来不难。

Numpy 中的 argmax

首先我们先来看一下 Numpy 中提供 argmax 函数，它重要的特点就是在有多个最大值的情况下，只返回第一个出现的最大值的位置。

In [1]: import numpy as np
In [2]: a = [1, 2, 9, 2, 5, 6, 9]
In [3]: np.argmax(a)
Out[3]: 2

如果需要返回所有最大值的位置的话，还是要麻烦一下的：

In