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

Đặt giới hạn trên thanh màu của một đồ thị đường viền trong Matplotlib

Để đặt các giới hạn trên thanh màu của ô đếm 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ụ.
  • Tạo x y điểm dữ liệu bằng cách sử dụng numpy.
  • Nhận dữ liệu bằng x y .
  • Nhận ma trận tọa độ từ các vectơ tọa độ.
  • Khởi tạo vmin vmax để đặt giới hạn trên thanh màu của một đường bao trong matplotlib.
  • Vẽ đường viền bằng cách sử dụng contourf () phương pháp.
  • Làm cho thanh màu bằng cách sử dụng vô hướng có thể lập bản đồ trong phạm vi vmin vmax .
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

from matplotlib import pyplot as plt
import numpy as np
from matplotlib.cm import ScalarMappable

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

x = np.arange(20)
y = np.arange(20)

data = x[:, None] + y[None, :]
X, Y = np.meshgrid(x, y)

vmin = 0
vmax = 15

fig, ax = plt.subplots()

qcs = ax.contourf(
      X, Y, data,
      vmin=vmin, vmax=vmax
)

fig.colorbar(
   ScalarMappable(norm=qcs.norm, cmap=qcs.cmap),
   ticks=range(vmin, vmax+5, 5)
)

plt.show()

Đầu ra

Đặt giới hạn trên thanh màu của một đồ thị đường viền trong Matplotlib