基于Python装饰器的向量化计算速度对比

timer是一个装饰器,功能是给被装饰的函数计时。如果要进一步了解装饰器的使用,点击此链接Python闭包函数和装饰器 sumOfLoop函数是常规的使用for进行循环遍历求和的方法; sumOfComprehension函数使用推导式得出新的列表,然后用内置sum函数求出列表的和; sumOfVectorization函数使用np.dot方法求出两个数据类型的为numpy.ndarray的对象的点积,两个向量a = [a1, a2,…, an]和b = [b1, b2,…, bn]的点积定义为:a·b=a1b1+a2b2+……+anbn。 np.random.rand()方法需要传入一个参数,例如传入参数为5,则返回一个数据类型为numpy.ndarray、长度为5、其中元素的值范围为0-1的对象,如下图所示:

np.random.rand()方法.png

from time import time
import numpy as np

def timer(func):
    def inner(*args,**kwargs):
        start = time()
        result = func(*args,**kwargs)
        end = time()
        usedTime = 1000 * (end - start)
        print("%s function used %.2f ms,return %.4f" %(func.__name__,usedTime,result))
        return result
    return inner

@timer
def sumOfLoop(np_array):
    result = 0
    for i in np_array:
        result += i * i
    return result

@timer
def sumOfComprehension(np_array):
    return sum([i * i for i in np_array])

@timer
def sumOfVectorization(np_array):
    return np.dot(np_array,np_array)

if __name__ == "__main__":
    print("计算小数平方和三种方法对比:")
    n = np.random.rand(3000000)
    a = sumOfLoop(n)
    print(a)
    sumOfComprehension(n)
    sumOfVectorization(n)
    print("计算整数平方和三种方法对比:")
    n = np.array(range(3000000)).astype('int64')
    sumOfLoop(n)
    sumOfComprehension(n)
    sumOfVectorization(n)

本文作者在2018年7月13日晚11点的运行结果如下:

计算小数平方和三种方法对比: sumOfLoop function used 1036.76 ms,return 999213.4882 sumOfComprehension function used 1103.75 ms,return 999213.4882 sumOfVectorization function used 2.00 ms,return 999213.4882 计算整数平方和三种方法对比: sumOfLoop function used 545.89 ms,return 8999995500000499712.0000 sumOfComprehension function used 718.86 ms,return 8999995500000499712.0000 sumOfVectorization function used 5.00 ms,return 8999995500000499712.0000

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏杂七杂八

numpy.nonzero()函数

官方文档如下: numpy.nonzero(a) Return the indices of the elements that are non-zero....

24530
来自专栏人工智能

如何为机器学习索引,切片,调整 NumPy 数组

具体在 Python 中,数据几乎被都被表示为 NumPy 数组。

75970
来自专栏mathor

搜索(6)

 题目大意是在一个nxn的方阵地图上,每一个方格都标记+号或者-号,要从A点到B点。题目要求移动路线要+-交替,问怎么移动从A到B才是最短路径?  同样...

14330
来自专栏数据结构与算法

21:角谷猜想

21:角谷猜想 总时间限制: 1000ms 内存限制: 65536kB描述 所谓角谷猜想,是指对于任意一个正整数,如果是奇数,则乘3加1,如果是偶数,则除以2...

40370
来自专栏程序生活

Tensorflow教程(十三) tf.Variable() 和tf.get_variable()1 简介2 区别3 实例

23230
来自专栏机器之心

入门 | 数据科学初学者必知的NumPy基础知识

选自TowardsDataScience 作者:Ehi Aigiomawu 机器之心编译 参与:李诗萌、路 本文介绍了一些 NumPy 基础知识,适合数据科学初...

28530
来自专栏从零开始的linux

python条件判断

if判断 a= 1 if a>10: print 'a大于10' print 'a小于10' a = -20 if a > 10: prin...

39090
来自专栏CVer

刷题笔记 | 剑指Offer 03 二维数组中的查找

本文主要讲解《剑指Offer》中第03题"二维数组中的查找",介绍题目、解决思路、解题步骤,并分别以C++和Python编程语言解答此题。

17830
来自专栏数据结构与算法

洛谷P3391 【模板】文艺平衡树(Splay)

题目背景 这是一道经典的Splay模板题——文艺平衡树。 题目描述 您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:翻转一个区...

378110
来自专栏Petrichor的专栏

numpy: np.random模块 探究(源码)

28120

扫码关注云+社区

领取腾讯云代金券