Thêm đường contour bổ sung trong contour plotting 2D Matplotlib

Để thêm đường contour bổ sung (extra contour lines) trong contour plotting 2D Matplotlib:

• Thiết lập kích thước figure và padding
• Tạo hàm f(x, y) tính z từ x, y
• Tạo dữ liệu x, y bằng numpy
• Tạo list levels bằng Numpy
• Vẽ contour plot bằng contour()
• Ghi nhãn contour và đặt title
• Hiển thị bằng show()

Ví dụ mã nguồn

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()

Kết quả

Contour plot với các đường mức (levels) từ -1.0 đến 1.5 bước 0.25, có ghi nhãn giá trị.

Tham số levels: Cho phép kiểm soát chính xác các đường contour được vẽ. Mặc định Matplotlib tự chọn ~7 levels.