python axis

import numpy as np

a=[   [ [1,5,1], [2,2,1] ],
      [ [5,6,1], [3,3,1] ],
      [ [3,3,3] , [4,4,4] ]   ]
a=np.array(a)
a1=np.mean(a,axis=(0,1))

print(a1)
#输出[3.         3.83333333 1.83333333]

(0,1)就是第2维以及更高维确定了之后,0,1维所有的变法都归为一组

把上面的数组想象成通道顺序维RGB的图像,第123维分别是(h,w,channel)

则相当于3张3行2列的rgb图

其中000,010,100,110,200,210归为1组

体现出来的效果就是对R通道所有位置的灰度求平均

猜你喜欢

转载自blog.csdn.net/weixin_45670020/article/details/114899189