Computer >> Máy Tính >  >> Lập trình >> Python

Thêm các đường đồng mức bổ sung bằng cách sử dụng vẽ đồ thị đường viền 2D Matplotlib

Để thêm các đường đồng mức phụ bằng cách sử dụng vẽ đồ thị đường viền 2D Matplotlib, chúng ta có thể thực hiện các bước sau -

  • Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô con.

  • Tạo e một hàm f (x, y) để lấy z điểm dữ liệu từ x y .

  • Tạo x y điểm dữ liệu bằng cách sử dụng numpy.

  • Tạo danh sách các cấp bằng Numpy.

  • Tạo biểu đồ đường viền bằng cách sử dụng contour () phương pháp.

  • Gắn nhãn cho ô đường viền và đặt tiêu đề cho ô.

  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

import matplotlib.pyplot as plt
import numpy as np

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

def f(x, y):
   return np.sin(x) ** 10 + np.cos(10 + y * x) * np.cos(x)

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

X, Y = np.meshgrid(x, y)
Z = f(X, Y)
levels = np.arange(-1.0, 1.5, 0.25)
CS = plt.contour(X, Y, Z, levels=levels)

plt.clabel(CS, inline=1, fontsize=10)

plt.title('levels = {}'.format(levels.tolist()))

plt.show()

Đầu ra

Thêm các đường đồng mức bổ sung bằng cách sử dụng vẽ đồ thị đường viền 2D Matplotlib