📅  最后修改于: 2023-12-03 15:26:57.681000             🧑  作者: Mango
在 Python 中,计算沿第三维数组的平均值很简单。需要使用 NumPy 库中的 mean() 函数,并传递一个参数 axis=2,表示计算第三维的平均值。
以下是一个简单的示例代码片段,演示了如何计算沿第三维数组的平均值:
import numpy as np
# 创建一个 3 * 2 * 4 的随机数组
arr = np.random.rand(3, 2, 4)
# 计算沿第三维数组的平均值
arr_mean = np.mean(arr, axis=2)
print(arr_mean)
输出:
array([[0.65705877, 0.46922 ],
[0.55141173, 0.49549503],
[0.53035953, 0.59861103]])
在上面的代码片段中,我们首先导入了 NumPy 库,然后创建了一个 3 * 2 * 4 的随机数组。接下来,我们使用 mean() 函数以 axis=2 作为参数来计算数组 arr 沿第三维的平均值。最后,我们打印出计算出来的平均值。
此外,我们还可以获取数组的 shape 属性来获取数组的维度信息,以帮助确定像第三维这样的特定维度适用于哪些计算。
print(arr.shape) # 输出 (3, 2, 4)
输出:
(3, 2, 4)
以上就是计算沿第三维数组的平均值的简单示例,希望能对您有所帮助!