国科大《Python科学计算与数据处理》作业2及参考答案(NumPy).docxVIP

1.依照课件3中的内容给出测算三角波(triangle_wave())y1,y2,y3,y4四种方式的计算速度与结果比较的代码，并对其运算显示结果。

代码：

importtime

importnumpyasnp

#基础三角波函数

deftriangle_wave(x,c,c0,hc):

x=x-int(x)#只取小数部分

ifx=c:

r=0.0

elifxc0:

r=x/c0*hc

else:

r=(c-x)/(c-c0)*hc

returnr

#生成测试数据

x=np.linspace(0,2,1000)

c_val,c0_val,hc_val=0.6,0.4,1.0

#1:列表推导式

start_time=time.time()

y1=np.array([triangle_wave(t,c_val,c0_val,hc_val)fortinx])

time_y1=time.time()-start_time

print(f方法1(列表推导式):{time_y1}秒)

#2:np.frompyfunc

start_time=ti