最小和argmin在numpy

时间:2014-09-27 15:56:51

标签: python numpy

给定一个数组我需要找到最小值和最小值的位置。 这可以使用

完成
>>> current_cost
array([ 2.54802261,  2.98627555,  0.23873749,  1.82511195,  1.35469083])
>>> current_cost.min()
0.23873748917821858
>>> current_cost.argmin()
2

此解决方案效率不高,因为它需要扫描列表两次。 有没有办法同时获得最小值和agrmin?

1 个答案:

答案 0 :(得分:15)

min_pos = current_cost.argmin()
min_val = current_cost[min_pos]