是的,你可以用
imshow
[matplotlib-doc]
为此:
data = [
[0,0,0,0,0,1,1,1,1,0],
[0,0,0,0,0,1,0,0,1,0],
[0,0,1,0,1,0,1,1,0,0],
[0,0,1,0,0,1,1,0,1,0],
[0,0,1,0,1,0,0,1,1,0],
[1,0,0,1,0,1,0,0,1,0],
[0,1,0,0,0,1,1,1,1,1],
[0,1,0,0,0,0,1,1,1,1],
[1,0,0,0,1,1,1,0,1,0],
[1,1,1,1,0,0,0,1,1,0]
]
import matplotlib.pyplot as plt
plt.imshow(data)
plt.show()
所以你只要把二维列表传给
imshow()
,这将生成以下图像:
如文档中所指定,您可以使用另一个颜色映射来使用其他颜色
0
和
1
,例如:
plt.imshow(data, cmap=plt.cm.bwr)
将生成相同的数据,但使用蓝-白-红颜色映射: