python中pyecharts绘制dataframe的热力图
时间: 2023-10-04 15:12:34 浏览: 357
Python绘制热力图示例
4星 · 用户满意度95%
可以使用pyecharts中的HeatMap模块来绘制DataFrame的热力图。首先需要将DataFrame转换为列表格式,然后将数据传递给HeatMap模块进行绘制。
示例代码如下:
```python
import pandas as pd
from pyecharts import HeatMap
# 构造DataFrame数据
data = {'A': [1, 2, 3, 4, 5],
'B': [5, 4, 3, 2, 1],
'C': [6, 7, 8, 9, 10],
'D': [10, 9, 8, 7, 6]}
df = pd.DataFrame(data)
# 将DataFrame转换为列表格式
data_list = []
for col in df.columns:
for i, value in enumerate(df[col]):
data_list.append([i, col, value])
# 绘制热力图
heatmap = HeatMap()
heatmap.add_xaxis(list(df.index))
heatmap.add_yaxis('', list(df.columns), data_list)
heatmap.render('heatmap.html')
```
执行以上代码后,会在当前目录下生成一个名为`heatmap.html`的文件,用浏览器打开即可查看生成的热力图。
阅读全文