矩阵列中的最小元素

时间:2015-03-04 20:05:48

标签: python numpy matrix minimum

我需要找到列中具有最大列总和的所有元素的最小值。 我做了以下事情:

创建随机矩阵

from numpy import *
a = random.rand(5,4)

然后计算每列的总和并找到最大元素的索引

c = a.sum(axis=0)
d = argmax(c)

然后我尝试在这一列中找到最小数字,但我的语法非常糟糕,我知道如何在当前索引的行中找到最小元素。

e = min(a[d])

但是如何更改列?

1 个答案:

答案 0 :(得分:5)

您可以按如下方式提取列的最小值(使用您指定的变量):

e=a[:,d].min()

请注意使用

a=min(a[:,d])

会让你脱离Numpy,放慢速度(感谢你指出@SaulloCastro)。