给定一个数组我需要找到最小值和最小值的位置。 这可以使用
完成>>> current_cost
array([ 2.54802261, 2.98627555, 0.23873749, 1.82511195, 1.35469083])
>>> current_cost.min()
0.23873748917821858
>>> current_cost.argmin()
2
此解决方案效率不高,因为它需要扫描列表两次。 有没有办法同时获得最小值和agrmin?
答案 0 :(得分:15)
min_pos = current_cost.argmin()
min_val = current_cost[min_pos]