在Python中实现argmax

Question

在Python中，argmax应该怎么实现呢？它应该尽可能高效，所以要能处理可迭代的对象。

可以有三种实现方式：

• 给定一组键值对，返回对应最大值的键
• 给定一组值，返回最大值的索引
• 给定一组键和一个函数f，返回使得f(key)最大的键

相关问题：

• numpy.argmax函数的错误行为（已解决，我的失误）
• 如何为容器对象实现__iter__(self)（Python）
• 从可迭代对象中获取第一个符合条件的项
• 我需要一个用于匹配mp3文件URL的href属性的正则表达式
• 我想在for循环中打印可迭代对象后添加新行
• 如何创建@iterator_cache装饰器？
• 我需要将一个需要按特定顺序排列的坐标整数列表转换为可迭代对象，以便小海龟可以前往这些坐标
• Python类型提示：我应该使用什么表示可以是任何可迭代对象的变量？

Answer 1

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

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

或者可以这样理解：

Answer 2

下面的代码是一种快速且符合Python风格的方法吗？

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

Answer 3

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

# 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)