python 读取excel文件并画出热力图的代码
时间: 2024-02-18 09:31:33 浏览: 224
读取excel文件,并进行绘图
3星 · 编辑精心推荐
以下是使用Python读取excel文件并画出热力图的代码示例:
```python
import pandas as pd
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt
# 读取excel文件
df = pd.read_excel('data.xlsx')
# 获取数据列
x_col = df.columns[0]
y_col = df.columns[1]
z_col = df.columns[2]
# 创建矩阵
matrix = np.zeros((len(df[x_col].unique()), len(df[y_col].unique())))
# 将数据填充到矩阵中
for i in range(len(df)):
x = df.iloc[i][x_col]
y = df.iloc[i][y_col]
z = df.iloc[i][z_col]
matrix[x][y] = z
# 画热力图
sns.heatmap(matrix, cmap='YlGnBu')
plt.show()
```
需要注意的是,上述代码使用了Pandas、NumPy、Seaborn和Matplotlib这些库,需要先安装这些库才能运行代码。此外,代码中的`data.xlsx`是一个Excel格式的数据文件,需要根据实际情况修改文件名和文件路径。
阅读全文