我知道我可以使用以下方式逐列获取numpy矩阵的最小值:
WidgetContainerComponent
但是,有什么方法可以有效地获得等效的索引/行而不是它们自己的值?例如:
my_array = [[10,2,3],[1,3,10],[20,19,0]]
我希望得到这样的东西:
[1,0,2] >>最小值的位置,而不是值。
答案 0 :(得分:1)
您可以改用argmin
>>> my_array = [[10, 2, 3], [1, 3, 10], [20, 19, 0]]
>>> a=np.array(my_array)
>>> np.argmin(a, axis=0)
array([1, 0, 2])