Numpy矩阵列最小值索引[重复]答案

【问题标题】：Numpy matrix column wise minimum value index [duplicate]Numpy矩阵列最小值索引[重复]
【发布时间】：2018-06-27 14:28:10
【问题描述】：

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

my_array.min(axis=0)

但是有什么方法可以有效地获得等效的索引/行而不是它们自身的值？例如：

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

我希望得到这样的东西：

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

【问题讨论】：

【解决方案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])

【讨论】：