numpy.argmax()

numpy.argmax(a, axis=None, out=None) 返回沿軸axis最大值的索引。 Parameters: a : array_like 數組 axis : int, 可選 默認情況下，索引的是平鋪的數組，否則沿指定的軸。 out : array, 可選 如果提供，結果以合適的形狀和類型被插入到此數組中。 Returns: index_array : ndarray of i

>>阅读原文<<