Numpy - 计算矩阵中哪些列最小的行?

时间:2017-11-16 03:49:37

标签: python numpy matrix

所以如果我有这样的矩阵:

[ 6 9 4 7]
[ 1 2 3 4]
[ 3 5 9 8] 
[ 5 7 2 4] 

我想知道第一列的最小值是多少次。例如,这里的答案是2,其中第2行和第3行的列为最小值。我知道我们可以使用this中的matrix.min(axis=1)找到最小值,我认为我们可以使用numpy.sum来计算行数......但是我怎么能真正知道从哪一列开始最低限度来自?我有一些简单的方法来实现这个numpy?

1 个答案:

答案 0 :(得分:0)

如果由于argmin的建议,它可以帮助其他人,那么这种方法可以用于第一列:

np.count_nonzero(np.argmin(matrix,axis=1))

不确定这种方法在大型矩阵或类似的东西上的性能然而..当然,除了检查第一列的最小值之外,这不起作用