最小和argmin在numpy

Don*_*beo 13 python numpy

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

>>> current_cost
array([ 2.54802261,  2.98627555,  0.23873749,  1.82511195,  1.35469083])
>>> current_cost.min()
0.23873748917821858
>>> current_cost.argmin()
2
Run Code Online (Sandbox Code Playgroud)

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

pv.*_*pv. 20

min_pos = current_cost.argmin()
min_val = current_cost[min_pos]
Run Code Online (Sandbox Code Playgroud)

  • 当您沿着一个轴获取多维数组的argmin时,这将如何工作?`min_pos = a.argmin(axis = 1)`,`min_vals =?` (3认同)