写一个相关性热力图的代码
时间: 2023-12-01 21:53:54 浏览: 67
热力图源码
以下是一个使用 Python 和 Seaborn 库绘制相关性热力图的示例代码:
```python
import seaborn as sns
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
# 创建一个随机的数据集
data = pd.DataFrame(np.random.rand(10, 10))
# 使用 Seaborn 绘制热力图
sns.heatmap(data.corr(), annot=True, cmap='coolwarm')
# 显示图形
plt.show()
```
在这个示例中,我们首先创建了一个随机的 $10 \times 10$ 数据集。然后,我们使用 Pandas 库计算数据集中每对列之间的相关性,并使用 Seaborn 库绘制了相关性热力图。
在 `sns.heatmap()` 函数中,我们设置 `annot=True` 以在图中显示相关性系数,并使用 `cmap='coolwarm'` 设置了色彩映射方案。最后,我们使用 `plt.show()` 函数显示图形。
阅读全文