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