python给二维图像添加colorbar

python给二维图像添加colorbar步骤如下

1. pandas读取表格每列数据
2. 采用plt.scatter方法进行绘图，添加cmap参数
3. 添加颜色条
4. 添加轴标签
5. 保存、显示图片

完整代码如下：

import pandas as pd
import matplotlib.pyplot as plt


# 读取Excel文件
excel_file_path = './epsilon_excel/2s.xlsx'  # 替换为你的Excel文件路径
df = pd.read_excel(excel_file_path)

# 假设我们要可视化的三列分别是 'Column1', 'Column2', 'Column3'
# 你可以替换这些列名为你Excel文件中的实际列名
column1 = df['X']
column2 = df['Y']
column3 = df['介电常数']

# 使用matplotlib绘制散点图
plt.scatter(column1, column2, c=column3, cmap='prism')  # c参数用于颜色映射，cmap参数用于颜色图谱

# 添加颜色条
plt.colorbar(label='Epsilon')

# 添加轴标签
plt.xlabel('X')
plt.ylabel('Y')
plt.title('Epsilon')
# 显示图形
plt.show()

缺点：
绘制出来的是散点图，看起来不是自己想要的那种效果。。。