numpy求多维(四维)数组均值 您所在的位置:网站首页 pythin求平均值 numpy求多维(四维)数组均值

numpy求多维(四维)数组均值

2024-01-21 23:22| 来源: 网络整理| 查看: 265

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 实验室设备网 版权所有