np.argmax的注意点
np.argmax函数
当参数是2维时，
axis=-1，代表纵轴，即选择纵轴方向最大的元素下标（即每一行的最大值）
axis=0，代表横轴，即选择横轴方向最大的元素下标（即每一列的最大值）
多维时，以此类推。
-- over --
np.argmax函数
当参数是2维时，
axis=-1，代表纵轴，即选择纵轴方向最大的元素下标（即每一行的最大值）
axis=0，代表横轴，即选择横轴方向最大的元素下标（即每一列的最大值）
多维时，以此类推。
-- over --