找出矩阵中最大的元素

Question

当提示输入矩阵中的行数，然后提示输入这些行的元素时，我需要找到矩阵中最大的元素并返回其位置（行和列）。在

例如，如果输入2行，如下所示：

[1 3 7] [4 8 1]

最大元素的位置将是（1，1），表示行索引1和列索引1。在

我有下面的代码，这半工半工得到的位置：

def main():
    matrix = []

    numRows = eval(input("Enter the number of rows in the list: "))
    for i in range(numRows):
        rows = input("Enter a row: ")
        items = rows.split()
        list = [eval(x) for x in items]
        matrix.append(list)

    return locateLargest(matrix)

def locateLargest(a):

    for i in range(len(a)):
        indexOfMaxRow = 0
        maxRow = max(a[i])

        for row in range(len(a)):
            if max(a[row]) > maxRow:
                maxRow = max(a[row])
                indexOfMaxRow = row

    for j in range(len(a)):
        indexOfMaxColumn = 0
        maxColumn = max(a[j])

        for column in range(len(a)):
            if max(a[column]) > maxColumn:
                maxColumn = max(a[column])
                indexOfMaxColumn = column

    print("The location of the largest element is at (", indexOfMaxRow, ", ", indexOfMaxColumn, ")")

main()

我认为我的代码在def locateLargest(a)下是错误的，因为在打印结果之前一切都正常。有人能告诉我是怎么回事吗？提前感谢您的任何意见！在

Answer 1

这太复杂了：

A = [[1, 3, 7],
     [4, 8, 1]]

def max_element(A):
    r, (c, l) = max(map(lambda t: (t[0], max(enumerate(t[1]), key=lambda v: v[1])), enumerate(A)), key=lambda v: v[1][1])
    return (l, r, c)

print max_element(A)

印刷品

^{pr2}$

Answer 2

您的locateLargest函数可以简单得多：

def locateLargest(matrix):
    largest_num = None 
    row = None
    col = None

    for row_idx, row in enumerate(matrix):
        for col_idx, num in enumerate(row):
            if num > largest_num or largest_num is None:
                largest_num = num
                row = row_idx
                col = col_idx

    return (largest_num, row, col)

此函数返回最大的数字，以及列和行索引（基于0的索引）。在

示例：

^{pr2}$

Answer 3

如果可以使用numpy，那么可以很容易地找到数组中最大值的索引：

import numpy as np
na = np.array(a)
indices = np.where(na == na.max())