3维numpy数组理解和np.argmax(a, axix=2)