Advertisement

matrix中计算并获取其每行与每列的最大值及其索引

阅读量:

矩阵每行/每列最大值

对数组aa进行行方向最大值计算的函数表达式为:np.amax(aa, axis= 1),而用于获取列方向最大值的对应表达式则为:np.amax(aa, axis= 0)

矩阵每行/列最大值的索引

行方向操作采用**np.argmax(aa, axis= 1)函数实现,该函数用于获取沿行方向的最大值索引;列方向操作则通过np.argmax(aa, axis= 0)**完成,其作用是确定沿列方向的最大值对应的位置。

示例:

  1. 矩阵中每一行的最大数值及其对应的位置信息。
复制代码
 aa =   [[0.25, 0.97, 0.17, 0.04, 0.49],

    
    [0.97, 0.94, 0.67, 0.05, 0.92],
    
    [0.17, 0.67, 0.18, 0.03, 0.84],
    
    [0.04, 0.74, 0.03, 0.  , 0.04],
    
    [0.49, 0.92, 0.95, 0.28, 0.  ]]
    
  
    
 np.argmax(aa, axis=1)
    
 Out: array([1, 0, 4, 1, 2], dtype=int64)
    
  
    
 np.amax(aa, 

全部评论 (0)

还没有任何评论哟~