Farbleiste in Matplotlib anzeigen

In diesem Tutorial werden wir diskutieren, wie man eine Farbleiste in einer Matplotlib-Figur in Python anzeigt.

Um eine Farbleiste zu erstellen, müssen wir die Funktion matplotlib.pyplot.colorbar() verwenden.

Der folgende Code zeigt ein einfaches Beispiel dafür.

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

s = plt.scatter(s_x, s_y, c=s_x, cmap="viridis")

c = plt.colorbar()

Im obigen Beispiel haben wir einen einfachen Farbbalken außerhalb des Plots erstellt. Die Colormap haben wir mit dem Parameter cmap angegeben.

Wir können auch die Achsen angeben, in denen wir die Farbleiste anzeigen möchten. Wenn wir möchten, können wir es über den Achsen des Diagramms hinzufügen.

Beispielsweise,

import random
import matplotlib.pyplot as plt

s_x = random.sample(range(0, 100), 20)
s_y = random.sample(range(0, 100), 20)

fig, ax = plt.subplots()
cax = fig.add_axes([0.27, 0.8, 0.5, 0.05])

im = ax.scatter(s_x, s_y, c=s_x, cmap="viridis")
fig.colorbar(im, cax=cax, orientation="horizontal")

Im obigen Beispiel haben wir die Funktion subplots() verwendet, um die Figuren- und Achsenobjekte abzurufen und damit die Achsen für die Farbleiste zu erstellen. Dies haben wir mit dem Parameter cax in der Funktion colorbar() angegeben.

Beachten Sie auch die Verwendung des Parameters orientation, der die Ausrichtung des endgültigen Farbbalkens verändert hat. Abgesehen davon können wir verschiedene Parameter verwenden, um die Form und Größe der Farbleiste zu steuern. Zum Beispiel kann shrink die Größe des Farbbalkens um einen kleinen Rand verkleinern, aspect, das ist das Verhältnis der Seiten des Balkens, und es gibt noch viele mehr.