Để thêm các vạch chia (ticks) tùy chỉnh cho colorbar, hãy làm theo các bước sau:
- Thiết lập kích thước figure và padding.
- Tạo dữ liệu x, y và z bằng NumPy.
- Hiển thị dữ liệu dưới dạng ảnh 2D bằng imshow().
- Tạo mảng ticks trong khoảng giá trị min và max của z bằng NumPy.
- Tạo colorbar cho instance ScalarMappable với tham số ticks=ticks.
- Hiển thị hình vẽ bằng show().
Ví dụ
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
x, y = np.mgrid[-1:1:100j, -1:1:100j]
z = (x + y) * np.exp(-5.0 * (x ** 2 + y ** 2))
plt.imshow(z, extent=[-1, 1, -1, 1])
ticks = np.linspace(z.min(), z.max(), 5, endpoint=True)
cb = plt.colorbar(ticks=ticks)
plt.show()
Kết quả
