(Tensorflow之十七）np.argmax（tf.argmax)的使用

一维数组

一维数组使用np.argmax返回的是数组内最大值的下标位置

import numpy as np

y1 = np.array([1,2,3,7,8,9])
y1 = np.argmax(y1)
print("the y1 is")
print(y1)

结果

the y1 is
5

二维数组

2.1 不指定axis

返回值是数组内的最大值的下标，以一维形式计算
例：

y2 = np.array([[1,2,3],[7,8,9]])
y2 = np.argmax(y2)
print("the y2 is")
print(y2)

结果

the y2 is
5

2.2 当axis = 0 时

返回值是各列中最大值的下标
例：

y2 = [[1,2,3],[7,8,9]]
y3 = np.argmax(y2,axis=0)
print("the y3 is")
print(y3)

结果

the y3 is
[1 1 1]

2.3 当axis = 1时

返回值是各行中最大值的下标
例：

y2 = [[1,2,3],[7,8,9]]
y4 = np.argmax(y2,axis=1)
print("the y4 is")
print(y4)

结果

the y4 is
[2 2]