Python如何求出矩阵中的最大值最小值以及所在的位置
Python如何求出矩阵中的最大值最小值及其所在的位置,以及当行的所有元素
·
想着记录下来吧,代码小白,一点一点来。
题目简述:有个50x53的矩阵G,第53列是每一行的总和,目的是求出第53列的最小值,并输出最小值所在行的所有元素。
因为要根据第52列来找出最小值,调用min()函数便能求出最小值;min = G[:,52].min()
结果为13561.0,发现对应的是矩阵的第10行,接下来我们的目标就是输出第十行的所有元素;
调用函数argmin()找出最小值所在的行记为变量c,c = np.argmin(G[:,52])
然后用for循环遍历第c行的所有元素,path = np.zeros([1,52]) for i in range(51): path[0][i]=G[c][i]
结果为:
求最大值类似,如有其他更好方法,还望多多交流!
更多推荐
已为社区贡献1条内容
所有评论(0)