scipy.sparse.csr_matrix.

argmax

csr_matrix.argmax(axis=None, out=None)

返回沿轴的最大元素的索引。

隐式零元素也被考虑在内。如果有多个最大值，则返回第一个出现的索引。

参数：:

axis{-2, -1, 0, 1, None}, 可选

计算 argmax 的轴。如果为 None（默认值），则返回扁平化数据中最大元素的索引。

outNone, 可选

此参数仅出于 NumPy 兼容性原因出现在签名中。不要传入除默认值以外的任何内容，因为此参数未使用。

返回：:

indnumpy.matrix 或 int

最大元素的索引。如果为矩阵，则沿 axis 的大小为 1。