python绘图:散点图,密度图并保存成图片

该博客展示了如何使用Python进行数据拟合和点密度估计。通过读取Excel文件,运用scipy的optimize库进行线性拟合,并利用matplotlib进行数据可视化,包括散点图、拟合线和颜色映射,以揭示数据分布特征。
摘要由CSDN通过智能技术生成

 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()

  • 2
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值