Python绘制热力图主要使用Seaborn和Matplotlib。Seaborn语法简洁,适合快速展示相关性矩阵,如sns.heatmap(data, annot=True, cmap='coolwarm');Matplotlib通过imshow()提供更细粒度控制,适用于自定义场景。结合pandas的corr()方法可直观分析变量相关性,Seaborn更实用,Matplotlib更灵活,按需选用。
Python绘制热力图主要通过可视化库实现,常用方法集中在Matplotlib和Seaborn上,操作简便且效果直观。数据通常以二维数组或DataFrame形式传入,适合展示相关性矩阵、密度分布等场景。
基本用法如下:
import seaborn as sns import matplotlib.pyplot as plt import pandas as pd示例数据
data = pd.DataFrame([[1, 5, 6], [4, 3, 2], [7, 8, 9]], columns=['A', 'B', 'C']) sns.heatmap(data, annot=True, cmap='coolwarm', center=0) plt.show()
可通过imshow()函数实现:
import matplotlib.pyplot as plt import numpy as npdata = np.random.rand(5, 5) plt.imshow(data, cmap='viridis') plt.colorbar() plt.xticks(range(5), ['X1', 'X2', 'X3', 'X4', 'X5']) plt.yticks(range(5), ['Y1', 'Y2', 'Y3', 'Y4', 'Y5']) plt.show()
步骤包括:
基本上就这些,Seaborn最实用,Matplotlib更灵活,按需选择即可。