Đặt giới hạn colorbar cho contour plot trong Matplotlib

Các bước

1. Thiết lập kích thước hình và padding.
2. Tạo dữ liệu x, y bằng numpy.
3. Tính dữ liệu lưới.
4. Tạo ma trận tọa độ bằng meshgrid().
5. Khởi tạo vmin, vmax để giới hạn colorbar.
6. Vẽ contour bằng contourf() với vmin, vmax.
7. Tạo colorbar từ ScalarMappable trong khoảng vmin-vmax.
8. Gọi show().

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

Kết quả