python画图颜色代码_Python matplotlib - 绘图中的颜色代码+ ve和-ve值

您可以创建一个三维数组，为每个像素分配颜色代码。因此，如果你想要黑色和白色，你将分别通过(0,0,0)和(1,1,1)。像这样的东西应该工作：def show_as_image(sample):

bitmap = sample.reshape((13, 8))

bitmap_colored = np.zeros((13,8,3))

bitmap_colored[bitmap>=0] = [1,1,1] # black for values greater or equal to 0

bitmap_colored[bitmap<0] = [0,0,0] # white for values less than 0

plt.figure()

plt.imshow(bitmap_colored, interpolation='nearest')

plt.show()

例如：>>> sample = np.random.randint(low=-10,high=10,size=(1,104))

>>> show_as_image(sample)

将输出这样的东西：