用np.mean写求三维数组的平均值

以下是使用np.mean计算三维数组平均值的示例代码：

import numpy as np

# 创建一个3x3x3的三维数组
a = np.array([[[1, 2, 3], [4, 5, 6], [7, 8, 9]],
              [[10, 11, 12], [13, 14, 15], [16, 17, 18]],
              [[19, 20, 21], [22, 23, 24], [25, 26, 27]]])

# 计算整个数组的平均值
print(np.mean(a)) # 输出：14.0

# 沿着第一个轴计算平均值
print(np.mean(a, axis=0)) # 输出：[[10. 11. 12.]
                          #      [13. 14. 15.]
                          #      [16. 17. 18.]]

# 沿着第二个轴计算平均值
print(np.mean(a, axis=1)) # 输出：[[ 4.  5.  6.]
                          #      [13. 14. 15.]
                          #      [22. 23. 24.]]

# 沿着第三个轴计算平均值
print(np.mean(a, axis=2)) # 输出：[[ 2.  5.  8.]
                          #      [11. 14. 17.]
                          #      [20. 23. 26.]]