matplotlib.pyplot 使用 cmap参数 修改配色

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
import pandas as pd import seaborn as sns import matplotlib.pyplot as plt data = pd.read_excel(r"D:桌面/shujukuE.xlsx", sheet_name="Sheet3") ds = pd.DataFrame(data) dataset = ds.copy() data.drop(["materials", "contain H", "contain C", "contain N", "contain P", "contain O", "contain S", "contain Te", "contain Se", "contain F", "contain Cl", "contain Br", "contain I", "jili", ], axis=1, inplace=True) 首先计算出相关系数 cor = data.corr(method='pearson') print(cor) # 输出相关系数 rc = {'font.sans-serif': 'SimHei', 'axes.unicode_minus': False} sns.set(font_scale=0.4, rc=rc) # 设置字体大小 设置热力图颜色配色 colors = "YlGnBu" # 颜色配置" color = colors.split('_') for i in color: i = i.strip() print(i) sns.heatmap(cor, annot=False, # 显示相关系数的数据 center=0.5, # 居中 fmt='.2f', # 只显示两位小数 linewidth=0, # 设置每个单元格的距离 #linecolor='blue', # 设置间距线的颜色# vmax=1.0, vmin=-0.5, # 设置数值最小值和最大值 xticklabels=True, yticklabels=True, # 显示x轴和y轴 square=True, # 每个方格都是正方形 cbar=True, # 绘制颜色条 cmap=f'{i}', # 设置热力图颜色 ) plt.xticks(fontsize=6) plt.yticks(fontsize=6) plt.savefig("D:/桌面/影响因素热力图颜色{i}.png", dpi=600) # 保存图片,分辨率为600 plt.ion() # 显示图片,这个可以方便后面自动关闭 plt.show() plt.pause(0.5)这段代码中我想把得到的相关性数据保存为Excel,应该再怎么优化代码
07-12
你可以使用`to_excel`方法将相关系数数据保存为Excel文件。以下是修改后的代码示例: ```python import pandas as pd import seaborn as sns import matplotlib.pyplot as plt data = pd.read_excel(r"D:桌面/shujukuE.xlsx", sheet_name="Sheet3") ds = pd.DataFrame(data) dataset = ds.copy() data.drop(["materials", "contain H", "contain C", "contain N", "contain P", "contain O", "contain S", "contain Te", "contain Se", "contain F", "contain Cl", "contain Br", "contain I", "jili"], axis=1, inplace=True) # 首先计算出相关系数 cor = data.corr(method='pearson') print(cor) # 输出相关系数 # 保存相关系数数据为Excel文件 cor.to_excel("D:/桌面/相关系数.xlsx", sheet_name="相关系数") rc = {'font.sans-serif': 'SimHei', 'axes.unicode_minus': False} sns.set(font_scale=0.4, rc=rc) # 设置字体大小 # 设置热力图颜色配色 colors = "YlGnBu" # 颜色配置" color = colors.split('_') for i in color: i = i.strip() print(i) sns.heatmap(cor, annot=False, # 显示相关系数的数据 center=0.5, # 居中 fmt='.2f', # 只显示两位小数 linewidth=0, # 设置每个单元格的距离 #linecolor='blue', # 设置间距线的颜色# vmax=1.0, vmin=-0.5, # 设置数值最小值和最大值 xticklabels=True, yticklabels=True, # 显示x轴和y轴 square=True, # 每个方格都是正方形 cbar=True, # 绘制颜色条 cmap=f'{i}', # 设置热力图颜色 ) plt.xticks(fontsize=6) plt.yticks(fontsize=6) plt.savefig(f"D:/桌面/影响因素热力图颜色{i}.png", dpi=600) # 保存图片,分辨率为600 plt.ion() # 显示图片,这个可以方便后面自动关闭 plt.show() plt.pause(0.5) ``` 在这个示例中,`cor.to_excel("D:/桌面/相关系数.xlsx", sheet_name="相关系数")`将相关系数数据保存为名为"相关系数.xlsx"的Excel文件,其中数据将保存在名为"相关系数"的工作表中。你可以根据需要调整文件路径和工作表名称。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值