python时间&内存计算

君莫笑xxx

已于 2024-04-10 21:29:22 修改

阅读量321

点赞数 7

分类专栏：程序文章标签： python 开发语言

于 2024-04-10 16:56:35 首次发布

本文链接：https://blog.csdn.net/u014426939/article/details/137602746

版权

程序专栏收录该内容

32 篇文章 1 订阅

订阅专栏

本文介绍了如何在Python中使用time模块测量函数执行时间，以及如何通过memory_profiler和psutil库来分别计算内存占用。作者通过示例展示了如何在代码中插入这些工具的使用方法。

摘要由CSDN通过智能技术生成

记录python时间内存计算操作

时间计算采用time模块
内存计算分别采用memory_profiler和psutil

1.占用时间计算：time

import time

def funT(t):
    for i in range(t*10):
        pass


if __name__=='__main__':
    t=10
    begin=time.time()
    funT(t)
    end=time.time()
    print('时间：{}s'.format(end - begin))
#时间：3.0994415283203125e-06s

2.占用内存计算：memory_profiler

#方法1
from memory_profiler import profile #pip 安装


@profile
def FunS(n):
    [1]*(10**n)


if __name__=='__main__':
    n=7
    FunS(n) 

#Line #    Mem usage    Increment  Occurrences   Line Contents
#=============================================================
#     8     18.4 MiB     18.4 MiB           1   @profile
#     9                                         def DemoSpace(n):
#    10     94.7 MiB     76.3 MiB           1       [1]*(10**n)

3. 占用内存空间：psutil

import psutil
import os

def show_info():
    #计算消耗内存
    pid = os.getpid()
    # 模块名比较容易理解：获得当前进程的pid
    p = psutil.Process(pid)
    # 根据pid找到进程，进而找到占用的内存值
    info = p.memory_info()
    memory = info.rss / 1024 / 1024
    return memory

def FunS(n):
    [1]*(10**n)


if __name__=='__main__':
    n = 7
    start_memory = show_info()
    FunS(n)  # 需要计算的算法步骤
    end_memory = show_info()
    print(f'一共占用{end_memory - start_memory}MB')
    #一共占用76.3046875MB