วิธีสร้างแผนที่ความร้อนใน Python ที่มีตั้งแต่สีเขียวถึงสีแดง (แมทพล็อตลิบ)

ในการสร้างแผนที่ความร้อนใน Python ที่มีตั้งแต่สีเขียวจนถึงสีแดง เราสามารถทำตามขั้นตอนต่อไปนี้ -

• กำหนดขนาดรูปและปรับช่องว่างภายในระหว่างและรอบๆ แผนผังย่อย
• สร้างพจนานุกรมสำหรับสีต่างๆ
• สร้าง colormap จากกลุ่มการทำแผนที่เชิงเส้นโดยใช้ LinearSegmentedColormap .
• สร้างร่างและชุดแผนย่อย
• สร้างข้อมูลสุ่มที่มีมิติ 5☓5
• สร้างพล็อตสีเทียมด้วยตารางสี่เหลี่ยมที่ไม่ธรรมดา
• สร้างแถบสีสำหรับ ScalarMappable ตัวอย่าง *แมปได้* .
• หากต้องการแสดงรูป ให้ใช้ show() วิธีการ

ตัวอย่าง

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

ผลลัพธ์