np.vstack() 将数组沿垂直方向堆叠起来
np.hstack() 将数组沿水平方向堆叠起来
numpy.argmax(a, axis=None, out=None) 返回沿轴axis最大值的索引号。axis=0按列输出，axis=1按行输出。