python将表格数据画成格网图
时间: 2023-09-08 22:11:30 浏览: 111
您可以使用 `matplotlib` 库绘制格网图,具体实现方法如下:
```python
import pandas as pd
import matplotlib.pyplot as plt
# 读取 Excel 文件
df = pd.read_excel('your_file_name.xlsx')
# 绘制格网图
fig, ax = plt.subplots()
cmap = plt.cm.get_cmap('YlOrRd') # 设置颜色表
im = ax.imshow(df, cmap=cmap) # 绘制格网图
plt.colorbar(im) # 添加 colorbar
plt.show()
```
其中,`your_file_name.xlsx` 是您的 Excel 文件名。上述代码将读取 Excel 表格数据并将其绘制成格网图并显示出来。您可以根据需要自行调整颜色表、坐标轴标签、标题等参数。
如果您的 Excel 数据表格中存在 `nan` 值,可以使用 `cmap.set_bad(color='black')` 将其设为黑色,示例如下:
```python
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
# 读取 Excel 文件
df = pd.read_excel('your_file_name.xlsx')
# 绘制格网图
fig, ax = plt.subplots()
cmap = plt.cm.get_cmap('YlOrRd') # 设置颜色表
cmap.set_bad(color='black') # 将 nan 值设为黑色
im = ax.imshow(df, cmap=cmap) # 绘制格网图
plt.colorbar(im) # 添加 colorbar
plt.show()
```
另外,如果您需要调整格网图的尺寸和比例,可以使用 `figsize` 参数和 `aspect` 参数,例如:
```python
fig, ax = plt.subplots(figsize=(8, 6))
im = ax.imshow(df, cmap=cmap, aspect='auto') # 设置宽高比例为自动调整
plt.colorbar(im)
plt.show()
```
您可以根据需要自行调整尺寸和比例。
阅读全文