【coo格式稀疏矩阵相乘】

coo格式两个稀疏矩阵相乘的方式

import numpy as np
from scipy.sparse import coo_matrix

# 定义两个 COO 格式稀疏矩阵 A 和 B
data_A = np.array([1, 2, 3, 4, 5, 6])
row_A = np.array([0, 0, 1, 1, 2, 2])
col_A = np.array([0, 2, 1, 2, 0, 1])
A = coo_matrix((data_A, (row_A, col_A)), shape=(3, 3))

data_B = np.array([7, 8, 9, 10, 11, 12])
row_B = np.array([0, 0, 1, 1, 2, 2])
col_B = np.array([0, 1, 1, 2, 0, 2])
B = coo_matrix((data_B, (row_B, col_B)), shape=(3, 3))

# 稀疏矩阵相乘
C = A.dot(B)

# 输出结果
print(C.toarray())