numpy 使用技巧

numpy.argmax 返回一個array的最大值所在行，列 ；返回第一個值web numpy.argmax(x,axis=,out=) >>> a = np.arange(6).reshape(2,3) >>> a array([[0, 1, 2], [3, 4, 5]]) >>> np.argmax(a) 5 >>> np.argmax(a, axis=0) array([1, 1, 1]

>>阅读原文<<