给定一个稀疏csr矩阵MS，我如何迭代MS的每一行？

Question

我有一个名为MS的稀疏矩阵，它是从5000x5000的密集矩阵派生而来的，我希望迭代它的每一行(以列表或其他形式)，以便可以对其执行其他步骤。例如：查找每行的总计数。

我已经尝试在网上查看了几个API，但我对此仍然是新手，并且在解释/理解它们时遇到了困难。

有没有什么办法可以让我在python中遍历稀疏csr矩阵MS的每一行？提前感谢您的任何帮助/意见。

Answer 1

你可以使用numpy

import numpy

new_MS = numpy.asarray(MS)

这将创建一个一维数组。

然后，您可以使用以下命令重塑列和行的尺寸：

new_ms = numpy.reshape(new_MS,(no_of_row,no_of_col))

这将创建一个指定行和列的二维数组

然后像二维数组上的简单c语言一样遍历列。