将列添加到稀疏矩阵
Posted
技术标签:
【中文标题】将列添加到稀疏矩阵【英文标题】:Add column to a sparse matrix 【发布时间】:2017-06-15 17:18:36 【问题描述】:当我执行以下代码时,我得到一个备用矩阵:
import numpy as np
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])
sp = csr_matrix((data, (row, col)), shape=(3, 3))
print(sp)
(0, 0) 1
(0, 2) 2
(1, 2) 3
(2, 0) 4
(2, 1) 5
(2, 2) 6
我想在这个稀疏矩阵中添加另一列,所以输出是:
(0, 0) 1
(0, 2) 2
(0, 3) 7
(1, 2) 3
(1, 3) 7
(2, 0) 4
(2, 1) 5
(2, 2) 6
(2, 3) 6
基本上,我想添加另一个值为 7、7、7 的列。
【问题讨论】:
看看here 【参考方案1】:@Paul Panzer's
链接中使用的sparse.hstack
是最简单的。
In [760]: sparse.hstack((sp,np.array([7,7,7])[:,None])).A
Out[760]:
array([[1, 0, 2, 7],
[0, 0, 3, 7],
[4, 5, 6, 7]], dtype=int32)
sparse.hstack
并不复杂;它只是调用bmat([blocks])
。
sparse.bmat
获取所有块的coo
属性,将它们与适当的offself 连接,并构建一个新的coo_matrix
。
在这种情况下它加入
In [771]: print(sp)
(0, 0) 1
(0, 2) 2
(1, 2) 3
(2, 0) 4
(2, 1) 5
(2, 2) 6
In [772]: print(sparse.coo_matrix(np.array([7,7,7])[:,None]))
(0, 0) 7
(1, 0) 7
(2, 0) 7
同时将最后的列号更改为3
。
In [761]: print(sparse.hstack((sp,np.array([7,7,7])[:,None])))
(0, 0) 1
(0, 2) 2
(1, 2) 3
(2, 0) 4
(2, 1) 5
(2, 2) 6
(0, 3) 7
(1, 3) 7
(2, 3) 7
【讨论】:
.A
在做什么?
@chandresh .A
returns self as an ndarray object.以上是关于将列添加到稀疏矩阵的主要内容,如果未能解决你的问题,请参考以下文章
SciPy NumPy 和 SciKit-learn ,创建一个稀疏矩阵