老卫带你学---Python用插值法绘制平滑曲线(包含3d)

原图:
在这里插入图片描述
平滑处理后:

在这里插入图片描述

代码实现如下:

# 1. 随机构造数据
import numpy as np
x = range(10)
y = np.random.randint(10,size=10)

# 2. 绘制原图
import matplotlib as mpl
import matplotlib.pyplot as plt
%matplotlib inline # jupyter notebook显示绘图
from scipy.interpolate import spline # 借助scipy库
plt.plot(x, y)
plt.show()

# 3. 绘制平滑曲线
from scipy.interpolate import spline
 # 插值法,50表示插值个数,个数>=实际数据个数,一般来说差值个数越多,曲线越平滑
x_new = np.linspace(min(x),max(x),50) 

y_smooth = spline(x, y, x_new)

plt.plot(x_new, y_smooth)

plt.show()

以上经过平滑后的曲线可能和原图差距很大,这个主要看数据本身的规律性。如果数据本身比较杂乱无章,如下:
在这里插入图片描述
则平滑后为:
在这里插入图片描述
平滑后曲线和原图还是差得很远的,因此该方法慎用,不是万能的。

3维光滑曲线

如果要对三维直线进行光滑处理,比如(x,y,z)。
我们只需要在2维的基础上加入对z的光滑处理就可以了

def add_line_smooth(results):   ###results就是我们的三维点的列表
    x_result=[]
    y_result=[]
    z_result=[]
    for i in results:
        i=list(i)
        x_result.append(i[0])
        y_result.append(i[1])
        z_result.append(i[2])
    xnew=np.linspace(max(x_result),min(x_result),30)
    y_result=np.array(y_result)
    ysmooth=spline(x_result,y_result,xnew)
    zsmooth=spline(x_result,z_result,xnew)
    ax.plot3D(xnew,ysmooth,zsmooth,c='r')
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值