numpy求多维(四维)数组均值 | 您所在的位置:网站首页 › pythin求平均值 › numpy求多维(四维)数组均值 |
numpy求均值
我们知道通过“np.mean(a)”可以求得N为矩阵左右元素的均值,但是有时候我们要求多维数组特定维度的均值我们该怎么办呢? 假如我们要求多维数组hsi_target除第二维之外的其他维度的均值。其中hsi_target的维度信息为:(1,31,1392,1300),我们要得到一个长度为31的一维数组。 已知:np.mean(a,axis=0)为每列的均值,np.mean(a,axis=1)为求每一行的均值,因此np.mean(a,axis=n)为求第n维的均值,因此我们可以利用嵌套的方法求多维数组某些维度的均值。 dome hsi_target = target.cpu().numpy() hsi_output = output.cpu().numpy() mean_target = np.mean(np.mean(np.mean(hsi_target,0),1),1) mean_output = np.mean(np.mean(np.mean(hsi_output,0),1),1) print(mean_target.shape) print(mean_output.shape) #输出结果为:(31,) #(31,) |
今日新闻 |
推荐新闻 |
专题文章 |
CopyRight 2018-2019 实验室设备网 版权所有 |