python - 乘以 CSR 稀疏矩阵的列
问题描述
我有以下稀疏的 CSR 矩阵
from scipy.sparse import csr_matrix
row = np.array([0, 0, 1, 2, 2, 2])
col = np.array([0, 2, 2, 0, 1, 2])
data = np.array([1, 2, 3, 4, 5, 6])
matrix = csr_matrix((data, (row, col)), shape=(3, 3))
而这个数组
weights = np.asarray([3, 6, 9])
我想做以下事情
matrix.toarray() * weights
但没有将稀疏矩阵转换为密集数组。
我试着做
matrix * weights
但这确实是点积,而不是我想要的列乘法。
知道如何在不将整个 CSR 矩阵转换为密集数组的情况下实现这一点吗?
解决方案
对于这个矩阵:
>>> matrix.A
array([[1, 0, 2],
[0, 0, 3],
[4, 5, 6]])
标准乘法默认为点积(与 一样matrix @ weights
):
>>> matrix * weights
array([21, 27, 96])
虽然有一个逐点乘法函数。这使列相乘:
>>> matrix.multiply(weights).A
array([[ 3, 0, 18],
[ 0, 0, 27],
[12, 30, 54]])
您还可以使用它通过广播来增加行数:
>>> matrix.multiply(weights[:, np.newaxis]).A
array([[ 3, 0, 6],
[ 0, 0, 18],
[36, 45, 54]])
推荐阅读
- relayjs - 当父记录为根时,Relay Modern updater ConnectionHandler.getConnection() 返回 undefined
- java - 错误:原因:io.netty.handler.codec.http2.Http2Exception:流的窗口大小溢出:0
- r - 检查两个不同的变量是否匹配
- dask - 异步计算 dask 数组块(Dask + FastAPI)
- javascript - 尝试使用 jquery 宽度更改图像
- carousel - react-bootstrap 如何控制轮播设置,控件,指示器,自转,间隔时间
- wpf - WPF如何在后面的代码中检索绑定属性
- typo3 - 如何在 TYPO3 的 Extbase 前端插件中获取根线/面包屑
- rest - vscode / sublime text 在保存时发出发布请求
- android - 使用 Kotlin 在房间中进行一对多