如果我想要一个列表中的最大值,我可以只写max(list),但如果我还需要最大值的索引呢?
我可以这样写:
maximum=0
for i,value in enumerate(List):
if value>maximum:
maximum=value
index=i
但我觉得很乏味。
如果我写:
List.index(max(List))
然后它将迭代该列表两次。
有没有更好的办法?
如果我想要一个列表中的最大值,我可以只写max(list),但如果我还需要最大值的索引呢?
我可以这样写:
maximum=0
for i,value in enumerate(List):
if value>maximum:
maximum=value
index=i
但我觉得很乏味。
如果我写:
List.index(max(List))
然后它将迭代该列表两次。
有没有更好的办法?
当前回答
max([(v,i) for i,v in enumerate(my_list)])
其他回答
这个答案比@Escualo快33倍,假设列表非常大,并且它已经是一个np.array()。我不得不减少测试运行的次数,因为测试要查看10000000个元素,而不仅仅是100个。
import random
from datetime import datetime
import operator
import numpy as np
def explicit(l):
max_val = max(l)
max_idx = l.index(max_val)
return max_idx, max_val
def implicit(l):
max_idx, max_val = max(enumerate(l), key=operator.itemgetter(1))
return max_idx, max_val
def npmax(l):
max_idx = np.argmax(l)
max_val = l[max_idx]
return (max_idx, max_val)
if __name__ == "__main__":
from timeit import Timer
t = Timer("npmax(l)", "from __main__ import explicit, implicit, npmax; "
"import random; import operator; import numpy as np;"
"l = np.array([random.random() for _ in xrange(10000000)])")
print "Npmax: %.2f msec/pass" % (1000 * t.timeit(number=10)/10 )
t = Timer("explicit(l)", "from __main__ import explicit, implicit; "
"import random; import operator;"
"l = [random.random() for _ in xrange(10000000)]")
print "Explicit: %.2f msec/pass" % (1000 * t.timeit(number=10)/10 )
t = Timer("implicit(l)", "from __main__ import explicit, implicit; "
"import random; import operator;"
"l = [random.random() for _ in xrange(10000000)]")
print "Implicit: %.2f msec/pass" % (1000 * t.timeit(number=10)/10 )
我电脑上的结果:
Npmax: 8.78 msec/pass
Explicit: 290.01 msec/pass
Implicit: 790.27 msec/pass
max([(v,i) for i,v in enumerate(my_list)])
我建议一个非常简单的方法:
import numpy as np
l = [10, 22, 8, 8, 11]
print(np.argmax(l))
print(np.argmin(l))
希望能有所帮助。
我列了一些大清单。一个是列表,一个是numpy数组。
import numpy as np
import random
arrayv=np.random.randint(0,10,(100000000,1))
listv=[]
for i in range(0,100000000):
listv.append(random.randint(0,9))
使用jupyter笔记本的%%time功能,我可以比较各种事情的速度。
2秒:
%%time
listv.index(max(listv))
54.6秒:
%%time
listv.index(max(arrayv))
6.71秒:
%%time
np.argmax(listv)
103女士:
%%time
np.argmax(arrayv)
Numpy的数组非常快。
使用Python的内置库,这非常简单:
a = [2, 9, -10, 5, 18, 9]
max(xrange(len(a)), key = lambda x: a[x])
这告诉max找到列表中最大的数字[0,1,2,…], len(a)],使用自定义函数lambda x: a[x],这表明0实际上是2,1实际上是9,等等。