2012-02-13 38 views

回答

8

getrow(i)返回1×n的CSR矩阵,它有一个indices属性,它给出了相应的值的行指数在data属性。 (我们知道形状是1个XN,所以我们没有对付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是一个空数组,则会引发异常。

-1

使用numpy.argmax(或scipy.argmax它是同一回事)

index_of_maximum = scipy.argmax(R.getrow(i).data) 
相关问题