在Numpy中，argmax 已经被定义，但我需要 a…