scipy.sparse.csr_array.

argmax

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

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

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

参数:

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

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

出None, 可选

此参数仅出于与 NumPy 兼容的原因存在于签名中。除了默认值外，不要传递任何内容，因为这个参数不会被使用。

返回:

indnumpy.matrix 或 int

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