python - numpy 数组中维度的平均值

Question

我的 numpy 数组(名称:数据)具有以下大小:(10L,3L,256L,256L)。它有 10 张图像，每张有 3 个颜色 channel (RGB)，每张图像大小为 256x256 像素。

我想计算所有 10 个图像的每个颜色 channel 的平均像素值。如果我使用 numpy 函数 np.mean(data)，我会收到所有像素值的平均值。使用 np.mean(data, axis=1) 返回大小为 (10L, 256L, 256L) 的 numpy 数组。

Answer 1

如果我正确理解您的问题，您需要一个数组，其中包含三个图像中每个图像的每个 channel 的平均值。 (即形状数组 (10,3) )(如果这不正确，请在评论中告诉我，我可以编辑此答案)

如果您使用的 numpy 版本高于 1.7，您可以将多个轴作为元组传递给 np.mean

mean_values = data.mean(axis=(2,3))

否则，您必须先展平阵列才能使其成为正确的形状。

mean_values = data.reshape((data.shape[0], data.shape[1], data.shape[2]*data.shape[3])).mean(axis=2)