1.用列表+循环实现,并包装成函数
def pySum(n): a = list(range(n)) b = list(range(0,5*n,5)) c = [] for i in range(len(a)): c.append(a[i]**2+b[i]**3) return (c) print(pySum(10))
2.用numpy实现,并包装成函数
import numpy a = numpy.arange(10) b = numpy.arange(0,50,5) c = a+b print(a,b,c)
输出
3.对比两种方法实现的效率,给定一个较大的参数n,用运行函数前后的timedelta表示。
import sys from datetime import datetime def pySum(n): a = list(range(n)) b = list(range(0,5*n,5)) c = [] for i in range(len(a)): c.append(a[i]**2+b[i]**3) return (c) print(pySum(10)) import numpy def npSum(n): a = numpy.arange(10) b = numpy.arange(0,50,5) c = a+b return c print(npSum(10)) from datetime import datetime start = datetime.now() pySum(10000) delta = datetime.now()-start print(delta) from datetime import datetime start = datetime.now() npSum(10000) delta = datetime.now()-start print(delta)