python key=operator.itemgetter(1)) 如何工作？

Question

我有一个矩阵，我需要找到最大元素及其数量。 如何在没有运算符（使用 for）的情况下重写它？

    for j in range(size - 1):
        i, val = max(enumerate(copy[j::, j]), key=operator.itemgetter(1))
        copy = change_rows(copy, j, i)
        P = change_rows(P, j, i) 

实际上，也许您可以解释一下这个字符串的含义？

i, val = max(enumerate(copy[j::, j]), key=operator.itemgetter(1))

Answer 1

让我们分解这条线。

i, val = max(enumerate(copy[j::, j]), key=operator.itemgetter(1))

首先， enumerate()在copy[j::,j]上创建一个迭代器，生成索引-值对。 例如，

>>> for i, val in enumerate("abcd"):
...     print(i, val)
...
0 a
1 b
2 c
3 d

接下来， max() function 用于查找序列中的最大项目。 但是我们希望它针对copy[j::,j]的值，而不是我们也从enumerate()获得的索引。 指定key=operator.itemgetter(1)告诉max()查看(i,val)对并找到具有最大val的一对。

使用np.argmax()可能会更好，特别是因为val未使用。

>>> import numpy as np

>>> for j in range(size - 1):
...     i = np.argmax(copy[j::, j])    # Changed this line.
        copy = change_rows(copy, j, i)
        P = change_rows(P, j, i)