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

Làm thế nào để vẽ đường viền và thang màu nhật ký trong Matplotlib?

Để vẽ đường bao và tỷ lệ nhật ký trong 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 ô phụ.
  • Khởi tạo một biến, N , để biết số lượng dữ liệu mẫu.
  • Tạo x, y, X, Y, Z1, Z2 z điểm dữ liệu bằng cách sử dụng numpy.
  • Tạo một hình và một tập hợp các tình huống phụ.
  • Vẽ đường viền bằng cách sử dụng contourf () phương pháp.
  • Tạo thanh màu cho một phiên bản có thể lập bản đồ vô hướng.
  • Để 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
from numpy import ma
from matplotlib import ticker, cm

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

N = 100
x = np.linspace(-3.0, 3.0, N)
y = np.linspace(-2.0, 2.0, N)

X, Y = np.meshgrid(x, y)

Z1 = np.exp(-X**2 - Y**2)
Z2 = np.exp(-(X * 10)**2 - (Y * 10)**2)

z = Z1 + 50 * Z2
z[:5, :5] = -1
z = ma.masked_where(z <= 0, z)
fig, ax = plt.subplots()

cs = ax.contourf(X, Y, z, locator=ticker.LogLocator(), cmap=cm.PuBu_r)
cbar = fig.colorbar(cs)

plt.show()

Đầu ra

Làm thế nào để vẽ đường viền và thang màu nhật ký trong Matplotlib?