最小和argmin在numpy

Question

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

完成

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

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

Answer 1

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