将列添加到稀疏矩阵

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 ,创建一个稀疏矩阵

将稀疏 scipy 矩阵加载到现有的 numpy 密集矩阵中

将稀疏矩阵转储到文件中

稀疏矩阵及其压缩格式