python - Pythonで密なベクトルとしてcoo_matrixから行を取得する方法は?

Question

私はPythonを初めて使用します。基本的な疎行列操作について教えてください。

1. 前もって行列全体を密にすることなく、疎行列から密な行ベクトルを抽出する方法は? coo_matrix.getrow()スパース表現のみを返します

2. 疎行列から行の割合 (たとえば 80%) をランダムに抽出する方法は? それらをトレーニングデータとして使用し、残りの割合をテストデータとして使用する必要があります。

前もって感謝します！

Answer 1

1. coo_matrix.getrow().todense()
2. たとえば、スライスをサポートする別のスパース表現を使用しますcsr_matrix。疎行列 A の場合、A[i] は i 番目の行になります。

例えば：

In [9]: from random import sample

In [10]: A = csr_matrix(...)

In [11]: n = A.shape[0]

In [12]: indices = sample(range(n), 4*n/5)

In [13]: A[indices].todense()