在matplotlib中绘制黑白二进制映射

Question

我正在使用python来模拟一些自动化模型,并且在matplotlib的帮助下,我正在生成如下所示的图.

我正在使用以下命令进行绘图:

ax.imshow(self.g, cmap=map, interpolation='nearest')

self.g二进制映射在哪里(0- >蓝色,1- >当前图中的红色).

但是,要在我的报告中包含这个,我希望情节是白色背景上的黑点而不是蓝色的红色.我该如何做到这一点？

Answer 1

您可以通过cmap关键字更改正在使用的颜色映射.颜色贴图'Greys'提供您想要的效果.您可以在scipy网站上找到可用地图的列表.

import matplotlib.pyplot as plt
import numpy as np

np.random.seed(101)
g = np.floor(np.random.random((100, 100)) + .5)

plt.subplot(211)
plt.imshow(g)
plt.subplot(212)
plt.imshow(g, cmap='Greys',  interpolation='nearest')
plt.savefig('blkwht.png')

plt.show()

这导致:

Answer 2

Yann的答案有一种替代方法可以让你更精细地控制.Matplotlib的imshow可以采用MxNx3矩阵,其中每个条目都是RGB颜色值 - 只需将它们设置为白色[1,1,1]或黑色[0,0,0].如果你想要三种颜色,很容易扩展这种方法.

import matplotlib.pyplot as plt
import numpy as np

# Z is your data set
N = 100
Z = np.random.random((N,N))

# G is a NxNx3 matrix
G = np.zeros((N,N,3))

# Where we set the RGB for each pixel
G[Z>0.5] = [1,1,1]
G[Z<0.5] = [0,0,0]

plt.imshow(G,interpolation='nearest')
plt.show()