So erstellen Sie ein Konturdiagramm in Matplotlib

Ein Konturdiagramm ist eine Art Diagramm, mit dem wir dreidimensionale Daten mithilfe von Konturen in zwei Dimensionen visualisieren können.

Sie können in Matplotlib ein Konturdiagramm erstellen, indem Sie die folgenden zwei Funktionen verwenden:

• matplotlib.pyplot.contour() - Erstellt Konturdiagramme.
• matplotlib.pyplot.contourf() - Erstellt gefüllte Konturdiagramme.

Die folgenden Beispiele zeigen, wie diese beiden Funktionen in der Praxis verwendet werden.

Beispiel 1: Konturdiagramm in Matplotlib

Angenommen, wir haben die folgenden Daten in Python:

import numpy as np

x = np.linspace(0, 5, 50)
y = np.linspace(0, 5, 40)

X, Y = np.meshgrid(x, y)
Z = np.sin(X*2+Y)*3 + np.cos(Y+5)

Wir können den folgenden Code verwenden, um ein Konturdiagramm für die Daten zu erstellen:

import matplotlib.pyplot as plt

plt.contour(X, Y, Z, colors='black')

Wenn eine einzelne Farbe für das Diagramm verwendet wird, stehen die gestrichelten Linien für negative Werte und die durchgezogenen Linien für positive Werte.

Eine Alternative besteht darin, eine Farbkarte mit dem Argument cmap anzugeben. Mit dem Argument levels können wir auch weitere Zeilen angeben, die im Plot verwendet werden sollen:

plt.contour(X, Y, Z, levels=30, cmap='Reds')

Wir haben uns für die cmap 'Reds' entschieden, aber eine vollständige Liste der Farbkartenoptionen finden Sie auf der Matplotlib-Dokumentationsseite.

Beispiel 2: Gefülltes Konturdiagramm in Matplotlib

Ein gefülltes Konturdiagramm ähnelt einem Konturdiagramm, außer dass die Zwischenräume zwischen den Linien gefüllt sind.

Der folgende Code zeigt, wie Sie mit der Funktion contourf() ein gefülltes Konturdiagramm für dieselben Daten erstellen, die wir im vorherigen Beispiel verwendet haben:

plt.contourf(X, Y, Z, cmap='Reds')

Wir können auch die Funktion colorbar() verwenden, um einen beschrifteten Farbbalken neben dem Plot hinzuzufügen:

plt.contourf(X, Y, Z, cmap='Reds')
plt.colorbar()

Weitere Matplotlib-Tutorials finden Sie hier.