Làm thế nào để tạo một bản đồ nhiệt trong Python có phạm vi từ màu xanh lá cây đến màu đỏ? (Matplotlib)

Để tạo bản đồ nhiệt trong Python có phạm vi từ màu xanh lá cây đến màu đỏ, 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 từ điển cho các màu khác nhau.
• Tạo bản đồ màu từ các phân đoạn ánh xạ tuyến tính bằng cách sử dụng Bản đồ định dạng tuyến tính .
• Tạo một hình và một tập hợp các ô phụ.
• Tạo dữ liệu ngẫu nhiên với thứ nguyên 5☓5.
• Tạo biểu đồ màu giả bằng lưới hình chữ nhật không thông thường.
• Tạo thanh màu cho ScalarMappable ví dụ, * có thể lập bản đồ * .
• Để 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 matplotlib.colors as colors
import numpy as np

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

cdict = {'red': ((0.0, 0.0, 0.0),
                (0.5, 1.0, 1.0),
                (1.0, 1.0, 0.7)),

        'green': ((0.0, 0.7, 0.7),
                  (0.5, 1.0, 1.0),
                  (1.0, 0.0, 0.0)),

         'blue': ((0.0, 0.0, 0.0),
                  (0.5, 1.0, 1.0),
                  (1.0, 0.0, 0.0))
         }
GnRd = colors.LinearSegmentedColormap('GnRd', cdict)

fig, ax = plt.subplots(1)

data = np.random.rand(5, 5)*6.-3.

p = ax.pcolormesh(data, cmap=GnRd, vmin=-5, vmax=5)

fig.colorbar(p, ax=ax)

plt.show()

Đầu ra