在Python中实现argmax

Question

如何在Python中实现argmax？它应该尽可能高效,因此它应该与iterables一起使用.

可以实施的三种方式:

• 给定可迭代的对返回对应于最大值的键
• 给定一个可迭代的值返回最大值的索引
• 给定一个可重复的键和一个函数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

基于Neil的答案,但专门针对带有多个参数的函数.

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

例如:

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

Answer 4

我发现这种方式更容易进行考虑argmax:说我们要计算argmax(f(y))其中y是从一个项目Y.因此,对于每个y我们想要计算f(y)并获得y最大值f(y).

argmax的这个定义是通用的,不像"给定一个可迭代的值返回最大值的索引"(并且它也很自然恕我直言).

并且..drumroll .. Python允许使用内置函数完成此操作max:

best_y = max(Y, key=f)

所以argmax_f(从接受的答案)是不必要的复杂和低效的恕我直言 - 这是一个复杂的内置版本max.此时所有其他类似argmax的任务应该变得清晰:只需定义一个合适的函数f.

Answer 5

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

或者类似地:

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