您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

python – 将稀疏数组中的元素与矩阵中的行相乘

5b51 2022/1/14 8:22:10 python 字数 1667 阅读 519 来源 www.jb51.cc/python

如果你有一个稀疏矩阵X: >> X = csr_matrix([[0,2,0,2],[0,2,0,1]]) >> print type(X) >> print X.todense() <class 'scipy.sparse.csr.csr_matrix'> [[0 2 0 2] [0 2 0 1]] 矩阵Y: >> print type(Y) >> print text_sco

概述

>> X = csr_matrix([[0,2,2],[0,1]])
>> print type(X)    
>> print X.todense()    
<class 'scipy.sparse.csr.csr_matrix'>
[[0 2 0 2]
 [0 2 0 1]]

矩阵Y:

>> print type(Y)
>> print text_scores
<class 'numpy.matrixlib.defmatrix.matrix'>
[[8]
 [5]]

…如何将X的每个元素乘以Y的行.例如:

[[0*8 2*8 0*8 2*8]
 [0*5 2*5 0*5 1*5]]

要么:

[[0 16 0 16]
 [0 10 0 5]]

我已经厌倦了这个,但显然它不起作用,因为尺寸不匹配:
Z = X.data * Y.

# Assuming that Y is 1D,might need to do Y = Y.A.ravel() or such...

# just to make the point that this works only with CSR:
if not isinstance(X,scipy.sparse.csr_matrix):
    raise ValueError('Matrix must be CSR.')

Z = X.copy()
# simply repeat each value in Y by the number of nnz elements in each row: 
Z.data *= Y.repeat(np.diff(Z.indptr))

这确实会产生一些临时性,但至少它是完全矢量化的,并且它不会使稀疏矩阵变得密集.

对于COO矩阵,等价物是:

Z.data *= Y[Z.row] # you can use np.take which is faster then indexing.

对于CSC矩阵,等价物将是:

Z.data *= Y[Z.indices]

总结

以上是编程之家为你收集整理的python – 将稀疏数组中的元素与矩阵中的行相乘全部内容,希望文章能够帮你解决python – 将稀疏数组中的元素与矩阵中的行相乘所遇到的程序开发问题。


如果您也喜欢它,动动您的小指点个赞吧

除非注明,文章均由 laddyq.com 整理发布,欢迎转载。

转载请注明:
链接:http://laddyq.com
来源:laddyq.com
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


联系我
置顶