python - 找到矩阵中第 i 列具有最大元素的行

Question

Python初学者，

我想创建一个如下方法:max(mat,i)= 矩阵 mat 的第 i 列中具有最大值的行。

比如我有一个矩阵a=[[1,2,3],[4,5,6],[7,8,9]]，那么最大值i=3 列是 9，因此 max(a,3)=[7,8,9]。

我想知道Python中有没有内置函数？

Answer 1

使用内置函数 max()

def matrix_max(matrix,column):
    temp = []
    for i in matrix:
        for j,item in enumerate(i):
            if j == column-1:
                temp.append(item)
    return max(temp)