脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|shell|

服务器之家 - 脚本之家 - Python - 你写的Python代码到底多快?这些测试工具了解了解

你写的Python代码到底多快?这些测试工具了解了解

2023-11-03 16:28啥都会一点的研究生 Python

当我们写完一个脚本或一个函数,首先能保证得到正确结果,其次尽可能的快(虽然会说Py这玩意咋整都慢,但有的项目就是得要基于Py开发)。 本期将总结几种获取程序运行时间的方法,极大的帮助对比不同算法/写法效率。 使用

当我们写完一个脚本或一个函数,首先能保证得到正确结果,其次尽可能的快(虽然会说Py这玩意咋整都慢,但有的项目就是得要基于Py开发)。

本期将总结几种获取程序运行时间的方法,极大的帮助对比不同算法/写法效率。

你写的Python代码到底多快?这些测试工具了解了解

使用系统命令

每个操作系统都有自己的方法来算程序运行的时间,比如在Windows PowerShell中,可以用 Measure-Command 来看一个Python文件的运行时间:

Measure-Command {python tutorial.py}

你写的Python代码到底多快?这些测试工具了解了解

在Ubuntu中,使用time命令:

time python tutorial.py

如果我们除了看整个 Python 脚本的运行时间外还想看看局部运行时间咋整?

使用 IPython 的 Magic Command

如果你使用过如Jupyter Notebook等工具会知道,他们用到了一个叫做 IPython 的交互式 Python 环境。

在 IPython 中,有一个特别方便的命令叫做 timeit。

对于某行代码的测量可以使用%timeit:

你写的Python代码到底多快?这些测试工具了解了解

对于某一个代码单元格的测量,可以使用%%timeit:

你写的Python代码到底多快?这些测试工具了解了解

使用timeit

如果不用IPython咋整,没关系,已经很厉害了,Python 有一个内置的timeit模块,可以帮助检测小段代码运行时间。

可以在命令行界面运行如下命令:

python -m timeit '[i for i in range(100)]'

使用 timeit 测量执行此列表推导式所需的时间,得到输出:

200000 loops, best of 5: 1.4 usec per loop

此输出表明每次计时将执行200000次列表推导,共计时测试了5次,最好的结果是1.4毫秒。

或者直接在Python中调用:

import timeit

print(timeit.timeit('[i for i in range(100)]', number=1))

对于更复杂的情况,有三个参数需要考虑:

  • stmt:待测量的代码片段,默认是 pass
  • setup:在运行 stmt 之前执行一些准备工作,默认也是 pass
  • number:要运行 stmt 的次数

比如一个更复杂的例子:

import timeit

# prerequisites before running the stmt
my_setup = "from math import sqrt"

# code snippet we would like to measure
my_code = '''
def my_function():
    for x in range(10000000):
        sqrt(x)
'''

print(timeit.timeit(setup=my_setup,
                    stmt=my_code,
                    number=1000))
# 6.260000000000293e-05

使用time模块

Python中内置的time模块相信都不陌生,基本的用法是在待测代码段的起始与末尾分别打上时间戳,然后获得时间差:

import time

def my_function():
    for i in range(10000000):
        pass
start = time.perf_counter()
my_function()
print(time.perf_counter()-start)
# 0.1179838

我经常使用time.perf_counter()来获取时间,更精确,在之前的教程中有提过。

time模块中还有一些其他计时选择:

  • time.timer():获取当前时间
  • time.perf_counter():计算程序的执行时间(高分辨率)
  • time.monotonic():计算程序的执行时间(低分辨率)
  • time.process_time():计算某个进程的CPU时间
  • time.thread_time():计算线程的CPU时间

假如我们需要在多个代码段测试运行时间,每个首尾都打上时间戳再计算时间差就有点繁琐了,咋整,上装饰器:

import time


def log_execution_time(func):
    def wrapper(*args, **kwargs):
        start = time.perf_counter()
        res = func(*args, **kwargs)
        end = time.perf_counter()
        print(f'The execution of {func.__name__} used {end - start} seconds.')
        return res

    return wrapper


@log_execution_time
def my_function():
    for i in range(10000000):
        pass


my_function()
# The execution of my_function used 0.1156899 seconds.

如上例所示,这样就使得代码非常干净与整洁。

原文地址:https://mp.weixin.qq.com/s?__biz=MzUzNjE1Nzc1MA==&mid=2247492393&idx=1&sn=923027d5047917da4da56000cfda8ecf

延伸 · 阅读

精彩推荐
  • Pythonpython中使用smtplib和email模块发送邮件实例

    python中使用smtplib和email模块发送邮件实例

    python脚本发邮件,一般会用到smtplib和email这两个模块。看看该模块怎么使用,先看smtplib模块。 smtplib模块定义了一个简单的SMTP客户端,可以用来在互联网上...

    脚本之家6432020-09-20
  • PythonPython web框架实现增加BasicAuth认证详解

    Python web框架实现增加BasicAuth认证详解

    这篇文章主要为大家详细介绍了Python如何在web框架中实现增加BasicAuth认证,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起了解一下...

    真的不能告诉你我的名字9142023-05-07
  • Python初探TensorFLow从文件读取图片的四种方式

    初探TensorFLow从文件读取图片的四种方式

    本篇文章主要介绍了初探TensorFLow从文件读取图片的四种方式,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧...

    Wayne201912922021-01-12
  • PythonPython统计列表元素出现次数的方法示例

    Python统计列表元素出现次数的方法示例

    这篇文章主要介绍了Python统计列表元素出现次数的方法示例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的...

    寸草心213011012021-09-30
  • PythonPytorch中的model.train() 和 model.eval() 原理与用法解析

    Pytorch中的model.train() 和 model.eval() 原理与用法解析

    pytorch可以给我们提供两种方式来切换训练和评估(推断)的模式,分别是:model.train() 和 model.eval(),这篇文章主要介绍了Pytorch中的model.train() 和 model.eval() 原...

    想变厉害的大白菜5472023-04-14
  • Pythonpython+pandas分析nginx日志的实例

    python+pandas分析nginx日志的实例

    下面小编就为大家分享一篇python+pandas分析nginx日志的实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    man8er10962021-02-07
  • PythonPython3 使用cookiejar管理cookie的方法

    Python3 使用cookiejar管理cookie的方法

    今天小编就为大家分享一篇Python3 使用cookiejar管理cookie的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    zjsxxzh11582021-05-09
  • Pythonpython操作RabbitMq的三种工作模式

    python操作RabbitMq的三种工作模式

    这篇文章主要为大家介绍了python操作RabbitMq的三种工作模式,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步早日升职加薪...

    Jeff的技术栈4082022-11-27