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

Vẽ đường trục hoặc gốc tọa độ cho biểu đồ contour trong Matplotlib

Để vẽ đường trục hoặc gốc tọa độ cho biểu đồ contour, ta dùng contourf(), axhline(y=0) và axvline(x=0).

  • Tạo điểm dữ liệu x, y, z bằng numpy.
  • Đặt thuộc tính trục bằng plt.axis('off').
  • Dùng contourf() với x, y, z.
  • Vẽ đường x=0 và y=0 màu đỏ.
  • Hiển thị bằng show().

Ví dụ

import numpy as np
import matplotlib.pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
x = np.linspace(-1.0, 1.0, 10)
x, y = np.meshgrid(x, x)
z = -np.hypot(x, y)
plt.axis('off')
plt.contourf(x, y, z, 10)
plt.axhline(0, color='red')
plt.axvline(0, color='red')
plt.show()

Kết quả

Vẽ đường trục hoặc gốc tọa độ cho biểu đồ contour trong Matplotlib