我需要找到列中具有最大列总和的所有元素的最小值。 我做了以下事情:
创建随机矩阵
from numpy import *
a = random.rand(5,4)
然后计算每列的总和并找到最大元素的索引
c = a.sum(axis=0)
d = argmax(c)
然后我尝试在这一列中找到最小数字,但我的语法非常糟糕,我知道如何在当前索引的行中找到最小元素。
e = min(a[d])
但是如何更改列?
答案 0 :(得分:5)
您可以按如下方式提取列的最小值(使用您指定的变量):
e=a[:,d].min()
请注意使用
a=min(a[:,d])
会让你脱离Numpy,放慢速度(感谢你指出@SaulloCastro)。