def argmax(pairs):
    return max(pairs, key=lambda x: x[1])[0]

def argmax_index(values):
    return argmax(enumerate(values))

def argmax_f(keys, f):
    return max(keys, key=f)
реализация argmax в Python