Python max-by function?

Question

Example:

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

This returns the maximum/biggest "chain_length" (an arbitrary function), but what I want is the i value for input that produces the biggest value.

Is there a convenient way to do that?

John La Rooy · Accepted Answer · 2012-07-05 02:10:43Z

30

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

John La Rooy

306k54 gold badges378 silver badges514 bronze badges

answered Jul 5, 2012 at 1:56

jamylak

134k30 gold badges238 silver badges240 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Didn't know about the key parameter. That's awesome! Thank you.