média ao longo da terceira dimensão Python

res = np.mean(X, axis=2)
Tremendous Enceladus