python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > numpy矩阵通道间求均值

如何对numpy 矩阵进行通道间求均值

作者:这就是算法

这篇文章主要介绍了如何对numpy 矩阵进行通道间求均值,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教

对numpy 矩阵进行通道间求均值

给定一个3x2x2得矩阵,得到一个2x2的矩阵,其中对应元素是通道间的均值

import numpy as np
a = np.array([
              [[1, 2],
              [3, 4]],
              [[5,6],
               [7,8]],
                [[1, 2],
              [3, 4]]
              ])
print(a.shape) # (3, 2, 2)
print(np.mean(a, axis=0))
"""
[[2.33333333 3.33333333]
 [4.33333333 5.33333333]]
"""
print(np.mean(a, axis=1))
"""
[[2. 3.]
 [6. 7.]
 [2. 3.]]
"""
print(np.mean(a, axis=2))
"""
[[1.5 3.5]
 [5.5 7.5]
 [1.5 3.5]]
"""

numpy求平均值的两种方法不一样

1、np.mean(x,axis=())

2、先reshape,再求均值(np.mean())

可是求出来的不一样,但是最后的三种均值的sum是相同的。

以后统一用一种方法,出bug在试一下另一种方法是否可行。

仅为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文