Hoe u een contourplot maakt in matplotlib

Een contourplot is een type plot waarmee we driedimensionale gegevens in twee dimensies kunnen visualiseren met behulp van contouren.

U kunt een contourplot maken in Matplotlib met behulp van de volgende twee functies:

• matplotlib.pyplot.contour() – Creëert contourplots.
• matplotlib.pyplot.contourf() – Creëert gevulde contourplots.

De volgende voorbeelden laten zien hoe u deze twee functies in de praktijk kunt gebruiken.

Voorbeeld 1: Contourplot in Matplotlib

Stel dat we de volgende gegevens in Python hebben:

 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)

We kunnen de volgende code gebruiken om een contourplot voor de gegevens te maken:

 import matplotlib.pyplot as plt

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

Wanneer er slechts één kleur wordt gebruikt voor de plot, vertegenwoordigen stippellijnen negatieve waarden en ononderbroken lijnen positieve waarden.

Een alternatief is om een kleurenkaart op te geven met behulp van het cmap- argument. We kunnen ook meer regels opgeven om in de plot te gebruiken met het argument niveaus :

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

We hebben ervoor gekozen om de cmap ‚Reds‘ te gebruiken, maar u kunt een volledige lijst met kleurenpaletopties vinden op de Matplotlib-documentatiepagina .

Voorbeeld 2: Gevulde contourplot in Matplotlib

Een gevuld contourpad is vergelijkbaar met een contourpad, behalve dat de ruimtes tussen de lijnen zijn gevuld.

De volgende code laat zien hoe u de functie contourf() gebruikt om een gevulde contourplot te maken voor dezelfde gegevens als in het vorige voorbeeld:

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

We kunnen ook de functie colorbar() gebruiken om een gelabelde kleurenbalk naast de plot toe te voegen:

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

Je kunt hier meer Matplotlib-tutorials vinden.