用Python实现argmax

Question

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

它可以通过三种方式来实施：

给定一个可迭代的对返回对应于最大值的键的键，给出了一个可迭代的值，返回了给定可迭代的键和函数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

下面的代码是一种快速和节奏式的方法吗？

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)