Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

การพล็อต Heatmap สำหรับ 3 คอลัมน์ด้วย Seaborn

ใช้ sns.heatmap() แสดง DataFrame 3x3 เป็น Heatmap

ขั้นตอน

  1. ตั้งค่าขนาด Figure
  2. สร้าง DataFrame 3 คอลัมน์ (a, b, c)
  3. วาด Heatmap ปิด Colorbar
  4. แสดงผล

ตัวอย่างโค้ด


import seaborn as sns
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

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

# DataFrame 3x3
df = pd.DataFrame(np.random.random((3, 3)), columns=["a", "b", "c"])

# Heatmap โดยไม่แสดง Colorbar
sns.heatmap(df, cbar=False)

plt.show()

การปรับแต่ง Heatmap


# แสดงค่าตัวเลข
sns.heatmap(df, annot=True, fmt=".2f", cbar=False)

# เปลี่ยน Colormap
sns.heatmap(df, cmap="RdYlBu_r", center=0.5)

# เพิ่ม Grid Lines
sns.heatmap(df, linewidths=0.5, linecolor='white')

# รวมทั้งหมด
sns.heatmap(df, annot=True, fmt=".2f", cmap="viridis", 
            linewidths=0.5, cbar_kws={"label": "Value"})

สำหรับข้อมูลขนาดใหญ่


# Correlation Matrix Heatmap
corr = df.corr()
sns.heatmap(corr, annot=True, cmap='coolwarm', center=0,
            square=True, fmt='.2f')

ผลลัพธ์

Heatmap 3x3 แสดงค่าสุ่ม 0-1 สีเข้ม = ค่าสูง สีอ่อน = ค่าต่ำ ไม่มี Colorbar