在 scipy 稀疏矩阵中选择行

Question

有没有办法在 scipy 稀疏矩阵中选择与某些给定索引对应的行？虚拟方法不起作用：

sparse.eye(3)[:2, :]

返回

TypeError: 'dia_matrix' object is not subscriptable

Answer 1

问这样的问题时，您应该说的不仅仅是“返回错误”。什么错误？这很重要。

但我会为你做这些工作：

In [143]: m =sparse.eye(3)
In [144]: m
Out[144]: 
<3x3 sparse matrix of type '<class 'numpy.float64'>'
    with 3 stored elements (1 diagonals) in DIAgonal format>
In [145]: m[:2,:]
...
TypeError: 'dia_matrix' object is not subscriptable

错误很严重。它告诉我们这种特殊的稀疏格式没有实现索引。对于常见的coo 格式，我们会遇到同样的错误。但是使用csr（或lil）格式，索引可以工作：

In [146]: m =sparse.eye(3, format='csr')
In [147]: m
Out[147]: 
<3x3 sparse matrix of type '<class 'numpy.float64'>'
    with 3 stored elements in Compressed Sparse Row format>
In [148]: m[:2,:]
Out[148]: 
<2x3 sparse matrix of type '<class 'numpy.float64'>'
    with 2 stored elements in Compressed Sparse Row format>
In [149]: _.A
Out[149]: 
array([[1., 0., 0.],
       [0., 1., 0.]])

在生成稀疏矩阵时，我喜欢将其显示为repr，它告诉我格式和大小。 print(m) (str) 以 coo 样式显示值。

sparse.eye 默认生成dia 格式，因为非零值都在一条对角线上。其他稀疏函数产生不同的默认格式。

---

dia 页面显示getrow 方法：

In [153]: sparse.eye(3).getrow(1)
Out[153]: 
<1x3 sparse matrix of type '<class 'numpy.float64'>'
    with 1 stored elements in Compressed Sparse Row format>

但请注意返回矩阵的格式 - csr，而不是 dia。这些格式具有易于相互转换的方法。一些操作为我们做了必要的转换。