Table of Contents
Install using pip
contour() method
parameter
Example
Output
contourf() function
Home Backend Development Python Tutorial Draw contour plots using Python Matplotlib

Draw contour plots using Python Matplotlib

Aug 30, 2023 pm 07:05 PM

Matplotlib is a free and open source plotting library in Python. It is used to create 2D graphics and plots by using python scripts. To use matplotlib functionality, we need to install the library first.

Install using pip

We can easily install the latest stable package of Matplotlib from PyPi by executing the following command in the command prompt.

1

pip install Matplotlib

Copy after login

You can install Matplotlib via conda using the following command -

1

conda install -c conda-forge matplotlib

Copy after login

Contour plot is used to visualize three-dimensional data in a two-dimensional surface by plotting constant z slices, called contours.

It is plotted with the help of the contour function (Z) which is a function of two inputs X and Y (X and Y axis coordinates).

1

Z = fun(x,y)

Copy after login

Matplotlib provides two functions plt.contour and plt.contourf to draw contour plots.

contour() method

matplotlib.pyplot. The outline() method is used to draw outline lines. It returns QuadContourSet. The following is the syntax of the function -

1

contour([X, Y,] Z, [levels], **kwargs)

Copy after login

parameter

  • [X,Y]: Optional parameter, indicating the coordinate of the Z value.

  • Z: The height value of the drawn outline.

  • levels: Used to determine the number and location of contour lines/areas.

Example

Let us take an example of drawing contour lines using numpy trigonometric functions.

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

import numpy as np

import matplotlib.pyplot as plt

 

def f(x, y):

   return np.sin(x) ** 10 + np.cos(10 + y * x) * np.cos(x)

 

xlist = np.linspace(-4.0, 4.0, 800)

ylist = np.linspace(-4.0, 4.0, 800)

 

# A mesh is created with the given co-ordinates by this numpy function

X, Y = np.meshgrid(xlist, ylist)

Z = f(X,Y)

 

fig = plt.figure()

ax = fig.add_axes([0.1, 0.1, 0.8, 0.8])

 

cp = ax.contour(X, Y, Z)

fig.colorbar(cp) # Add a colorbar to a plot

ax.set_title('Contour Plot')

 

ax.set_xlabel('x (cm)')

ax.set_ylabel('y (cm)')

plt.show()

Copy after login

Output

使用Python Matplotlib绘制等高线图

The f(x,y) function is defined using numpy trigonometric functions.

Example

Let’s take another example and draw contour lines.

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

import numpy as np

import matplotlib.pyplot as plt

 

def f(x, y):

    return np.sqrt(X**2 + Y**2)

 

xlist = np.linspace(-10, 10, 400)

ylist = np.linspace(-10, 10, 400)

 

# create a mesh

X, Y = np.meshgrid(xlist, ylist)

 

Z = f(X, Y)

 

fig = plt.figure(figsize=(6,5))

ax = fig.add_axes([0.1, 0.1, 0.8, 0.8])

 

cp = ax.contour(X, Y, Z)

ax.set_title('Contour Plot')

ax.set_xlabel('x (cm)')

ax.set_ylabel('y (cm)')

plt.show()

Copy after login

Output

使用Python Matplotlib绘制等高线图

The z function is the sum of the square roots of the x and y coordinate values. Implemented using the numpy.sqrt() function.

contourf() function

matplotlib.pyplot provides a method contourf() to draw filled contours. The following is the syntax of the function -

1

contourf([X, Y,] Z, [levels], **kwargs)

Copy after login

where,

  • [X,Y]: Optional parameter, indicating the coordinate of the Z value.

  • Z: The height value of the drawn outline.

  • levels: Used to determine the number and location of contour lines/areas.

Example

Let us take another example and use the contourf() method to draw a contour map.

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

import numpy as np

import matplotlib.pyplot as plt

 

xlist = np.linspace(-8, 8, 800)

ylist = np.linspace(-8, 8, 800)

 

X, Y = np.meshgrid(xlist, ylist)

Z = np.sqrt(X**2 + Y**2)

 

fig = plt.figure(figsize=(6,5))

ax = fig.add_axes([0.1, 0.1, 0.8, 0.8])

 

cp = ax.contourf(X, Y, Z)

fig.colorbar(cp) # Add a colorbar to a plot

ax.set_title('Filled Contours Plot')

#ax.set_xlabel('x (cm)')

ax.set_ylabel('y (cm)')

plt.show()

Copy after login

Output

使用Python Matplotlib绘制等高线图

Using the fig.colorbar() method, we add color to the drawing. The z function is the sum of the square roots of the x and y coordinate values.

Example

In this example, we will use the matplotlib.plt.contourf() method to plot a polar contour plot.

1

2

3

4

5

6

7

8

9

10

11

12

13

14

import numpy as np

import matplotlib.pyplot as plt

 

a = np.radians(np.linspace(0, 360, 20))

b = np.arange(0, 70, 10)

  

Y, X = np.meshgrid(b, a)

values = np.random.random((a.size, b.size))

  

fig, ax = plt.subplots(subplot_kw=dict(projection='polar'))

ax.set_title('Filled Contours Plot')

ax.contourf(X, Y, values)

  

plt.show()

Copy after login

Output

使用Python Matplotlib绘制等高线图

In all the above examples, we used the numpy.meshgrid() function to generate arrays of X and Y coordinates.

The above is the detailed content of Draw contour plots using Python Matplotlib. For more information, please follow other related articles on the PHP Chinese website!

Statement of this Website
The content of this article is voluntarily contributed by netizens, and the copyright belongs to the original author. This site does not assume corresponding legal responsibility. If you find any content suspected of plagiarism or infringement, please contact admin@php.cn

Hot AI Tools

Undresser.AI Undress

Undresser.AI Undress

AI-powered app for creating realistic nude photos

AI Clothes Remover

AI Clothes Remover

Online AI tool for removing clothes from photos.

Undress AI Tool

Undress AI Tool

Undress images for free

Clothoff.io

Clothoff.io

AI clothes remover

AI Hentai Generator

AI Hentai Generator

Generate AI Hentai for free.

Hot Article

R.E.P.O. Energy Crystals Explained and What They Do (Yellow Crystal)
2 weeks ago By 尊渡假赌尊渡假赌尊渡假赌
Repo: How To Revive Teammates
4 weeks ago By 尊渡假赌尊渡假赌尊渡假赌
Hello Kitty Island Adventure: How To Get Giant Seeds
3 weeks ago By 尊渡假赌尊渡假赌尊渡假赌

Hot Tools

Notepad++7.3.1

Notepad++7.3.1

Easy-to-use and free code editor

SublimeText3 Chinese version

SublimeText3 Chinese version

Chinese version, very easy to use

Zend Studio 13.0.1

Zend Studio 13.0.1

Powerful PHP integrated development environment

Dreamweaver CS6

Dreamweaver CS6

Visual web development tools

SublimeText3 Mac version

SublimeText3 Mac version

God-level code editing software (SublimeText3)

How to Use Python to Find the Zipf Distribution of a Text File How to Use Python to Find the Zipf Distribution of a Text File Mar 05, 2025 am 09:58 AM

This tutorial demonstrates how to use Python to process the statistical concept of Zipf's law and demonstrates the efficiency of Python's reading and sorting large text files when processing the law. You may be wondering what the term Zipf distribution means. To understand this term, we first need to define Zipf's law. Don't worry, I'll try to simplify the instructions. Zipf's Law Zipf's law simply means: in a large natural language corpus, the most frequently occurring words appear about twice as frequently as the second frequent words, three times as the third frequent words, four times as the fourth frequent words, and so on. Let's look at an example. If you look at the Brown corpus in American English, you will notice that the most frequent word is "th

How Do I Use Beautiful Soup to Parse HTML? How Do I Use Beautiful Soup to Parse HTML? Mar 10, 2025 pm 06:54 PM

This article explains how to use Beautiful Soup, a Python library, to parse HTML. It details common methods like find(), find_all(), select(), and get_text() for data extraction, handling of diverse HTML structures and errors, and alternatives (Sel

Image Filtering in Python Image Filtering in Python Mar 03, 2025 am 09:44 AM

Dealing with noisy images is a common problem, especially with mobile phone or low-resolution camera photos. This tutorial explores image filtering techniques in Python using OpenCV to tackle this issue. Image Filtering: A Powerful Tool Image filter

How to Work With PDF Documents Using Python How to Work With PDF Documents Using Python Mar 02, 2025 am 09:54 AM

PDF files are popular for their cross-platform compatibility, with content and layout consistent across operating systems, reading devices and software. However, unlike Python processing plain text files, PDF files are binary files with more complex structures and contain elements such as fonts, colors, and images. Fortunately, it is not difficult to process PDF files with Python's external modules. This article will use the PyPDF2 module to demonstrate how to open a PDF file, print a page, and extract text. For the creation and editing of PDF files, please refer to another tutorial from me. Preparation The core lies in using external module PyPDF2. First, install it using pip: pip is P

How to Cache Using Redis in Django Applications How to Cache Using Redis in Django Applications Mar 02, 2025 am 10:10 AM

This tutorial demonstrates how to leverage Redis caching to boost the performance of Python applications, specifically within a Django framework. We'll cover Redis installation, Django configuration, and performance comparisons to highlight the bene

How to Perform Deep Learning with TensorFlow or PyTorch? How to Perform Deep Learning with TensorFlow or PyTorch? Mar 10, 2025 pm 06:52 PM

This article compares TensorFlow and PyTorch for deep learning. It details the steps involved: data preparation, model building, training, evaluation, and deployment. Key differences between the frameworks, particularly regarding computational grap

How to Implement Your Own Data Structure in Python How to Implement Your Own Data Structure in Python Mar 03, 2025 am 09:28 AM

This tutorial demonstrates creating a custom pipeline data structure in Python 3, leveraging classes and operator overloading for enhanced functionality. The pipeline's flexibility lies in its ability to apply a series of functions to a data set, ge

Introduction to Parallel and Concurrent Programming in Python Introduction to Parallel and Concurrent Programming in Python Mar 03, 2025 am 10:32 AM

Python, a favorite for data science and processing, offers a rich ecosystem for high-performance computing. However, parallel programming in Python presents unique challenges. This tutorial explores these challenges, focusing on the Global Interprete

See all articles