本程序实现从txt文件中读取两列数据,然后进行三次样条插值,绘制出一条平滑的曲线。
需要解决的一些问题:切片方法:
x = a[:,0] # 取第一列数据
y = a[:,1] # 取第二列数据
另外一种使用 pandas 切片的方法:
# code:utf-8 Ubuntu
import matplotlib.pyplot as plt
from scipy import interpolate
import numpy as np
import matplotlib.font_manager as mpt
zhfont=mpt.FontProperties(fname='/usr/share/fonts/custom/msyh.ttf') #显示中文字体
#导入数据
file = 'data.txt'
a = np.loadtxt(file)
# 数组切片
x = a[:,0] # 取第一列数据
y = a[:,1] # 取第二列数据
# 进行样条插值
tck = interpolate.splrep(x,y)
xx = np.linspace(min(x),max(x),100)
yy = interpolate.splev(xx,tck,der=0)
print(xx)
# 画图
plt.plot(x,y,'o',xx,yy)
plt.legend(['true','Cubic-Spline'])
plt.xlabel('距离(cm)', fontproperties=zhfont) #注意后面