Funzione max-by Python?

Question

Esempio:

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

Questa operazione riporta la massima/più grande "chain_length" (una funzione arbitraria), ma quello che voglio è il valore i per l'input che produce il più grande valore.

C'è un modo conveniente per farlo?

Answer 1

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