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,应该再怎么优化代码
时间: 2023-11-08 20:04:04 浏览: 112
你可以使用`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文件,其中数据将保存在名为"相关系数"的工作表中。你可以根据需要调整文件路径和工作表名称。
阅读全文