Zuweisen von bestimmten Werten zum Matplotlib-Diagramm

Question

Ich habe zwei Werte in einem numply-Array. 0,1. Ich möchte sicherstellen, dass 1 immer schwarz ist und 0 immer grau ist, wenn ich das Array grafisch darstelle.

Wie kann ich dies in Matplotlib tun?

Danke

Answer 1

Unter der Annahme, dass Sie ein Bild Grundstück (plt.imshow()) darstellen möchten, können Sie eine colormap wie „gray_r“ auswählen können, die als unterste und schwarz als höchste Farbe weiß hat, wird grau in der Mitte sein. Wenn Sie jetzt das Bilddiagramm auf Werte zwischen -1 und 1 normalisieren, entspricht der Wert 0 des Arrays der Mitte der Farbpalette (grau) und 1 entspricht dem oberen Ende der Farbpalette (schwarz).

import matplotlib.pyplot as plt 
import numpy as np; np.random.seed(0) 

a = np.random.randint(0,2, size=(12,25)) 

plt.imshow(a, cmap="gray_r", vmin=-1, vmax=1) 

plt.show()