利用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))

我希望沿着最后一个轴(axis=2)将这两个稀疏矩阵组合成一个单一的稀疏矩阵。类似于：

s3 = sparse.COO(s1, s2)

Answer 1

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

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

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