在 Python 中实现 argmax

Question

argmax 在 Python 中应该如何实现？它应该尽可能高效，因此它应该与可迭代一起使用。

可以通过三种方式实现：

• 给定一个可迭代的对，返回对应于最大值的键
• 给定一个可迭代的值，返回最大值的索引
• 给定一个可迭代的键和函数 f，返回f(key) 最大的键

Answer 1

我修改了我找到的最佳解决方案：

# given an iterable of pairs return the key corresponding to the greatest value
def argmax(pairs):
    return max(pairs, key=lambda x: x[1])[0]

# given an iterable of values return the index of the greatest value
def argmax_index(values):
    return argmax(enumerate(values))

# given an iterable of keys and a function f, return the key with largest f(key)
def argmax_f(keys, f):
    return max(keys, key=f)

Answer 2

下面的代码是一种快速且Pythonic的方式吗？

idx_max = max(enumerate(x), key=lambda x:x[1])[0]

Answer 3

def argmax(lst):
     return lst.index(max(lst))

或类似地：

argmax = lambda lst: lst.index(max(lst)

Answer 4

我发现这种方式更容易考虑 argmax：假设我们要计算 argmax(f(y))，其中 y 是来自 Y 的项目。因此，对于每个 y，我们要计算 f(y) 并获得具有最大 f(y) 的 y。

argmax 的这个定义是通用的，不像“给定一个可迭代的值返回最大值的索引”（恕我直言，这也是很自然的）。

并且 ..drumroll.. Python 允许使用内置的 max 来做到这一点：

best_y = max(Y, key=f)

所以 argmax_f （来自已接受的答案）是不必要的复杂且低效的恕我直言 - 它是内置 max 的复杂版本。所有其他类似 argmax 的任务此时应该变得清晰：只需定义一个适当的函数 f 即可。

Answer 5

基于尼尔的答案，但专门用于采用多个参数的函数。

argmax = lambda keys, func: max(imap(lambda key: (func(*key), key), keys))[1]

例如：

argmax([(5, 2), (3, 3), (2, 5)], pow)
# (2, 5)