如何对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在试一下另一种方法是否可行。
仅为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。