Đặt hai biểu đồ imshow Matplotlib có cùng thang colormap

Để đặt hai biểu đồ imshow() Matplotlib có cùng thang colormap, chúng ta có thể thực hiện các bước sau:

Các bước

• Thiết lập kích thước figure và điều chỉnh padding giữa và xung quanh subplots.
• Tạo ma trận d1 và d2 bằng Numpy.
• Lấy ma trận kết quả để lấy giá trị max và min.
• Dùng amin và amax cho giá trị min và max.
• Tạo figure mới hoặc kích hoạt figure hiện có.
• Thêm '~.axes.Axes' vào figure như một phần của arrangement subplot, với nrows=1, ncols=2 tại index 1.
• Dùng phương thức imshow() với vmin và vmax, xác định phạm vi dữ liệu colormap bao phủ.
• Lặp lại bước 6 và 7 với dữ liệu.
• Để hiển thị figure, dùng phương thức show().

Ví dụ

import numpy as np
from matplotlib import pyplot as plt

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

data1 = np.random.randn(4, 4)
data2 = np.random.randn(4, 4)
resultant = np.array([data1, data2])
min_val, max_val = np.amin(resultant), np.amax(resultant)

fig = plt.figure()
ax = fig.add_subplot(1, 2, 1)
ax.imshow(data1, cmap="plasma", vmin=min_val, vmax=max_val)

ax2 = fig.add_subplot(1, 2, 2)
ax2.imshow(data2, cmap="plasma", vmin=min_val, vmax=max_val)

plt.show()

Kết quả