给定一个稀疏矩阵R
类型的scipy.sparse.coo_matrix
形状的1.000.000 x 70.000
我发现
row_maximum = max(R.getrow(i).data)
会给我第 i 行的最大值。
我现在需要的是值对应的索引row_maximum
.
有什么想法如何实现这一目标吗?
感谢您提前提供任何建议!
getrow(i)
返回一个 1 x n CSR 矩阵,其中有一个indices
属性给出了相应值的行索引data
属性。 (我们知道形状是 1 x n,所以我们不必处理indptr
属性。)所以这会起作用:
row = R.getrow(i)
max_index = row.indices[row.data.argmax()] if row.nnz else 0
我们必须处理以下情况row.nnz
分别为 0,因为row.data.argmax()
如果出现以下情况,将引发异常row.data
是一个空数组。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)