有效地改变scipy.spare.csr_matrix的维度

Question

我有一个函数,它需要csr_matrix并对其进行一些计算.

这些计算的行为要求该矩阵的形状是特定的(比如说NxM).

我发送的输入具有较少的列和确切的行数.

(例如它的形状=(A,B),其中A <N且B == M)

例如:我有对象 x

>>>x = csr_matrix([[1,2],[1,2]])
>>>x
(0, 0)  1
(0, 1)  2
(1, 0)  1
(1, 1)  2
>>>x.shape
(2, 2)

功能f:

def f(csr_mat):
    """csr_mat.shape should be (2,3)"""

然后我想做点什么x,所以它会成为y:

>>>y = csr_matrix([[1,2,0],[1,2,0]])
>>>y
(0, 0)  1
(0, 1)  2
(1, 0)  1
(1, 1)  2
>>>y.shape
(2, 3)

在这个例子中,x与y具有相同的无零的值,但y具有不同的形状.我想要的是有效地"扩展" x到一个新的维度,用零填充新的列.也就是说,考虑到x和new_shape=(2,3),它应该返回y.
我已经尝试过reshape:

x.reshape((2,3))

但后来我得到了:

NotImplementedError

我的第二个选择是创建csr_matrix具有不同形状的新东西:

z = csr_matrix(x,shape=(3,3))

但这也失败了:

NotImplementedError:未对csr_matrix实施重新整形.

编辑:使用csc_matrix带来了同样的错误.

有任何想法吗？

谢谢

Answer 1

在CSR格式中,所需的底层data,indices和indptr数组y与x矩阵的相同.您可以csr_matrix使用new 将它们传递给构造函数shape:

y = csr_matrix((x.data, x.indices, x.indptr), shape=(2, 3))

注意构造默认copy=False,所以这将共享data,indices以及indptr之间x和y.一些操作y将反映在x.你可以通过copy=True做x和y相互独立的.

如果您想要查看未记录csr_matrix的内部_shape属性,可以设置内部属性以使x数组具有您想要的形状:

x._shape = (2, 3)

这样做并没有什么优势.