从矩阵中获取具有最大总和值的列

Question

让我们说：

import numpy as np    
f=np.matrix("1 2; 3 4 ; 5 6")

是否可以从矩阵中检索具有最大列数的列数？怎么样？

Answer 1

你可以写：

>>> f.sum(axis=0).argmax()
1

所以列1总和为最大值。

澄清这样做：f.sum(axis=0)对矩阵f的列求和，返回矩阵matrix([[ 9, 12]])。然后argmax()用于查找此总和矩阵中最大值的索引。