对numpy 矩阵进行通道间求均值
给定一个3x2x2得矩阵,得到一个2x2的矩阵,其中对应元素是通道间的均值
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
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在试一下另一种方法是否可行。
仅为个人经验,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/qq_36321330/article/details/108508219