1.首先展示下测试数据:
2.poython读取文件,
import pandas as pd
import numpy as np
from scipy import optimize
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.colors import Normalize
from matplotlib import ticker
plt.style.use('seaborn-darkgrid')
dataset_path=r'C:\Users\Administrator\Desktop\blog\test.xlsx'
scatter = pd.read_excel(dataset_path,sheet_name= 'Sheet2')
x = scatter['x'].values.ravel() #真值
y = scatter['y'].values.ravel() #预测值
N = len(scatter['x'])
#绘制拟合线
x2 = np.linspace(-10,30)
y2 = x2
def f_1(x,A,B):
return A*x + B
A1,B1 = optimize.curve_fit(f_1,x,y)[0]
y3 = A1*x + B1
# Calculate the point density
xy = np.vstack([x,y])
z = gaussian_kde(xy)(xy)
norm = Normalize(vmin = np.min(z), vmax = np.max(z))
#开始绘图
fig,ax = plt.subplots(figsize=(8,5),dpi=200)
dian=plt.scatter(x,y,edgecolor=None,c=z,s=20,cmap='Spectral_r')
cb=plt.colorbar(ax=ax,drawedges=False)
cb.ax.locator_params(nbins=8)
cb.ax.set_yticklabels([0.005,0.010,0.015,0.020,0.025,0.030,0.035])#0,0.005,0.010,0.015,0.020,0.025,0.030,0.035
ax.plot(x2,y2,color='k',linewidth=1.5,linestyle='--')
ax.plot(x,y3,color='r',linewidth=2,linestyle='-')
fontdict1 = {"size":17,"color":"k",'family':'Times New Roman'}
ax.set_xlabel("x",fontdict=fontdict1)
ax.set_ylabel("y",fontdict=fontdict1)
ax.grid(True)
ax.set_xlim((30,65))
ax.set_ylim((25,65))
ax.set_xticks(np.arange(30,65, step=5))
ax.set_yticks(np.arange(25, 65, step=5))
plt.savefig(r'C:\Users\Administrator\Desktop\blog\test.png',dpi=900,bbox_inches='tight')
plt.show()