Python绘制热力图保存到excel中

在Python中，我们可以使用`seaborn`库来创建热力图，并通过`pandas`将数据保存到Excel文件中。以下是一个简单的步骤：

1. 首先，你需要安装所需的库，可以使用pip安装：

   pip install seaborn pandas matplotlib openpyxl

2. 导入必要的库：

import seaborn as sns
import pandas as pd
import matplotlib.pyplot as plt
from openpyxl import Workbook

3. 创建或准备数据，这里假设你有一个二维数组`data`：

data = [[你的数据填充在这里]]

4. 使用`seaborn`绘制热力图：

heatmap = sns.heatmap(data, annot=True, cmap='coolwarm')  # 设置颜色映射和标注
plt.title('Heatmap Title')  # 添加标题

5. 将图形显示出来（如果你需要保存图片，可以省略这一步）：

plt.show()

6. 将图表转换为DataFrame并保存到Excel文件：

df_heatmap = pd.DataFrame(data)
book = Workbook()  # 创建一个新的Excel工作簿
sheet = book.active  # 获取活动的工作表

# 将热力图数据写入Excel
for i in range(len(df_heatmap)):
    for j in range(len(df_heatmap.columns)):
        sheet.cell(row=i+1, column=j+1).value = df_heatmap.iloc[i][j]

# 保存Excel文件
book.save('heatmap.xlsx')