python+numpy按行求一个二维数组的最大值方法

问题描述： 给定一个二维数组，求每一行的最大值 返回一个列向量 如： 给定数组【1,2，3；4,5，3】 返回[3；5] import numpy as np x = np.array([[1,2,3],[4,5,3]]) # 先求每行最大值得下标 index_max = np.argmax(x, axis=1)# 其中，axis=1表示按行计算 print(index_max.shape) max = x[range(x.shape[0]), index_max] print(max) # 注意到这里返回的是行向量 # 这可以是一种通用的方法， # 其中range（）可以是一个列向量，表示