Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Tạo contourf tròn tròn trong Matplotlib

Để tạo matplotlib.pyplot.contourf hình tròn:

  • Thiết lập kích thước figure và padding
  • Tạo dữ liệu x, y, a, b, c bằng Numpy
  • Tạo figure và subplots
  • Vẽ contour plot bằng contourf()
  • Đặt tỷ lệ khung hình aspect=1
  • Hiển thị bằng show()

Ví dụ mã nguồn

import numpy as np
import matplotlib.pyplot as plt

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

x = np.linspace(-0.5, 0.5, 100)
y = np.linspace(-0.5, 0.5, 100)
a, b = np.meshgrid(x, y)
c = a ** 2 + b ** 2 - 0.2
figure, axes = plt.subplots()
axes.contourf(a, b, c)
axes.set_aspect(1)
plt.show()

Kết quả

Contourf hình tròn (đường tròn đồng tâm) nhờ set_aspect(1) đảm bảo tỷ lệ trục X=Y.

Tạo contourf tròn tròn trong Matplotlib

Giải thích: c = a² + b² - 0.2 là phương trình đường tròn bán kính √0.2. set_aspect(1) hoặc axis('equal') bắt buộc tỷ lệ đơn vị trục X=Y.