那python如何在sparse matrix中获取非零的行号和列好呢
时间: 2023-05-15 22:04:44 浏览: 166
python 判断矩阵中每行非零个数的方法
可以使用scipy.sparse库中的find函数来获取非零元素的行号和列号,示例代码如下:
import scipy.sparse as sp
# 创建一个稀疏矩阵
mat = sp.csr_matrix([[0, 0, 1], [2, 0, 0], [0, 3, 0]])
# 获取非零元素的行号和列号
rows, cols, values = sp.find(mat)
print(rows) # [0 1 2 1 2]
print(cols) # [2 0 1 0 1]
阅读全文