在 python 中使用稀疏包连接两个稀疏矩阵

Question

我在 pythons 稀疏包中有两个稀疏矩阵。见下文：

import sparse
total_coords1 = [(0,1,1,2), (0,0,2,3), (0,1,2,2)]
data1 = [1,1,1,1]
s1 = sparse.COO(total_coords1, data1, shape=(7, 5, 12))

total_coords2 = [(0,1,2,3), (0,1,1,2), (0,1,2,2)]
data2 = [2,2,2,2]
s2 = sparse.COO(total_coords1, data1, shape=(7, 5, 15))

我想将这两个稀疏矩阵合并成一个沿最后一个轴（轴=2）的稀疏矩阵。类似于：

s3 = sparse.COO(s1, s2)

Answer 1

由于您没有提到您想要连接的轴，我将假定轴 = 2，因为它是我们可以连接给定数组的唯一可能轴。

您可以使用 concatenate 函数来获得形状为 (7, 5, 27) 的单个稀疏矩阵：

s3 = sparse.concatenate([s1,s2], axis=2)

在 python 中使用稀疏包连接两个稀疏矩阵

Concatenating two sparse matrices using sparse package in python

python

numpy

scipy

sparse-matrix