numpy和纯python代码比较两向量相加

import numpy as np
import datetime

def pysum(n):
    a = list(range(n))
    b = list(range(n))
    c = []
    for i in range(len(a)):
        a[i] = i ** 2
        b[i] = i ** 3
        c.append(a[i] + b[i])
    return c# 使用Numpy实现两向量的加法



def npsum(n):
    a = np.arange(n) ** 2
    b = np.arange(n) ** 3
    c = a + b
    return c


if __name__ == '__main__':
    # print(pysum(10))
    # print(npsum(10))


    start = datetime.datetime.now()
    pysum(1000000)
    end = datetime.datetime.now()
    print("{end} - {start} = {sub}".format(start=start , end=end , sub = end - start))


    start = datetime.datetime.now()
    npsum(1000000)
    end = datetime.datetime.now()
    print("{end} - {start} = {sub}".format(start=start, end=end, sub=end - start))