块矩阵列明智最小值索引

时间:2018-06-27 06:24:31

标签: python numpy

我知道我可以使用以下方式逐列获取numpy矩阵的最小值:

WidgetContainerComponent

但是,有什么方法可以有效地获得等效的索引/行而不是它们自己的值?例如:

  

my_array = [[10,2,3],[1,3,10],[20,19,0]]

我希望得到这样的东西:

  

[1,0,2] >>最小值的位置,而不是值。

1 个答案:

答案 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])