Python max-by function?

Question

Ejemplo:

print max(chain_length(i) for i in xrange(1,10001)) 

Esto devuelve el máximo/mayor "chain_length" (una función arbitraria), pero lo que quiero es el valor i para la entrada que produce el valor más grande.

¿Hay una forma conveniente de hacerlo?

Answer 1

max(xrange(1, 10001), key=chain_length)