給定一個`scipy.sparse.coo_matrix`類型的矩陣如何確定每行最大值的索引和值？

Question

鑑於scipy.sparse.coo_matrix類型的稀疏矩陣R形狀1.000.000 x 70.000我想通了，

row_maximum = max(R.getrow(i).data) 

會給我第i行的最大值。

我現在需要的是對應的值row_maximum索引。

任何想法如何實現這一目標？

感謝您的任何建議提前！

Answer 1

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是一個空數組，則會引發異常。

Answer 2

使用numpy.argmax（或scipy.argmax它是同一回事）

index_of_maximum = scipy.argmax(R.getrow(i).data)