реализация argmax в Python

Я изменил лучшее решение, которое нашел:

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

Является ли следующий код быстрым и питоническим способом?

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

Мне показалось, что так проще думать об argmax: скажем, мы хотим вычислить argmax(f(y)), где y — элемент из Y. Итак, для каждого y мы хотим вычислить f(y) и получить y с максимальным f(y).

Это определение argmax является общим, в отличие от «данная итерация значений возвращает индекс наибольшего значения» (и это также вполне естественно ИМХО).

И ..барабанная дробь.. Python позволяет сделать именно это с помощью встроенного max:

best_y = max(Y, key=f)

Итак, argmax_f (из принятого ответа) ИМХО излишне сложный и неэффективный - это усложненная версия встроенного max. Все остальные задачи, подобные argmax, должны стать понятными на этом этапе: просто определите правильную функцию f.

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

или аналогично:

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

Основано на ответе Нила, но специализировано для функций, которые принимают несколько аргументов.

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

Например:

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