scipy.sparse.csr_array.

argmax

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

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

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

参数:

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

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

outNone, 可选

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

返回:

indnumpy.matrix 或 int

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