1、库安装
要绘制热力图,可以使用Python中的matplotlib库和seaborn库。
pip install matplotlib
pip install seaborn
2、示例代码
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
# 创建一个随机的2D数组
data = np.random.rand(10, 10)
# 绘制热力图
sns.heatmap(data)
plt.show()
这段代码会生成一个10x10的随机矩阵,并将其绘制成热力图。你可以根据自己的需求修改矩阵的大小和数据。
另外,你还可以通过调整cmap参数来改变热力图的颜色映射,例如:
sns.heatmap(data, cmap='coolwarm')
这样会使用冷暖色调的颜色映射。