numpy库的max 和 argmax

目录

定义矩阵的方法

max函数

argmax函数

---

 

定义矩阵的方法

X = np.matrix([[1,3,2],[4,1,2],[3,1,5]])

max函数

def max_test(X):

    y0 = np.max(X,axis = 0)
    y1 = np.max(X,axis = 1)
    y2 = np.max(X)
    print(X)
    print("每列的最大值：",y0)
    print("每行的最大值：\n",y1)
    print("整个矩阵的最大值：\n",y2)

输出：

[[1 3 2]
 [4 1 2]
 [3 1 5]]
每列的最大值： [[4 3 5]]
每行的最大值：
 [[3]
 [4]
 [5]]
整个矩阵的最大值：
 5

argmax函数

def argmax_test(X):
    y0 = np.argmax(X, axis=0)
    y1 = np.argmax(X, axis=1)
    y2 = np.argmax(X)
    print(X)
    print("每列的最大值索引：", y0)
    print("每行的最大值索引：\n", y1)
    print("整个矩阵的最大值索引：\n", y2)

输出：

[[1 3 2]
 [4 1 2]
 [3 1 5]]
每列的最大值索引： [[1 0 2]]
每行的最大值索引：
 [[1]
 [0]
 [2]]
整个矩阵的最大值索引：
 8