鸢尾花数据集绘制相关性热力图
时间: 2023-10-22 15:07:38 浏览: 246
鸢尾花数据集(txt格式,可直接使用)
3星 · 编辑精心推荐
可以使用seaborn库中的heatmap函数来绘制鸢尾花数据集的相关性热力图。具体步骤如下:
1. 导入需要的库和数据集
```python
import seaborn as sns
import pandas as pd
iris = sns.load_dataset('iris')
```
2. 计算相关系数矩阵
```python
corr_matrix = iris.corr()
```
3. 绘制热力图
```python
sns.heatmap(corr_matrix, annot=True, cmap='coolwarm')
```
这样就可以得到鸢尾花数据集的相关性热力图了。
阅读全文