media a lo largo de la matriz de tercer dimensión Python

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