如何在python中用颜色代码绘制矩阵稀疏模式？

Question

我正在使用 matplotlib.pyplot 中的 spy 从 scipy.sparse 像这样绘制 csc_matrix 的稀疏模式

>>> import scipy.sparse as sprs
>>> import matplotlib.pyplot as plt
>>> Matrix=sprs.rand(10,10, density=0.1, format='csc')
>>> plt.spy(Matrix)
>>> plt.show()

我想做同样的事情，但根据它们的大小为矩阵元素赋予颜色。有没有一种简单的方法可以让间谍做到这一点？如果没有，还有其他方法可以做到吗？

Answer 1

你可以使用imshow：

d=Matrix.todense()
plt.imshow(d,interpolation='none',cmap='binary')
plt.colorbar()

给出：