Python max-by函数?

mpe*_*pen 12 python

例:

print max(chain_length(i) for i in xrange(1,10001))
Run Code Online (Sandbox Code Playgroud)

这将返回最大/最大"chain_length"(任意函数),但我想要的是产生i最大值的输入值.

有没有方便的方法呢?

jam*_*lak 21

max(xrange(1, 10001), key=chain_length)
Run Code Online (Sandbox Code Playgroud)