绘制dataframe中某几列的热力图
时间: 2023-09-29 13:09:46 浏览: 159
Python读取txt某几列绘图的方法
要绘制dataframe中某几列的热力图,可以使用seaborn库中的heatmap函数。
示例代码如下:
```python
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
# 构造示例数据
data = {
'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]
}
df = pd.DataFrame(data)
# 绘制A列和B列的热力图
sns.heatmap(df[['A', 'B']].corr(), cmap='YlGnBu', annot=True)
plt.show()
```
这段代码会绘制出A列和B列的相关系数热力图,其中参数`cmap`指定了颜色映射,参数`annot`指定了是否在图中显示相关系数的值。你可以根据自己的需求修改数据和参数,绘制出符合你要求的热力图。
阅读全文