Python 檢測程序的運(yùn)行時(shí)間和占用內(nèi)存情況

在使用 Python 語言做算法、數(shù)據(jù)分析等等，通常程序員需要了解程序的效率和內(nèi)存使用的情況，從而更好地了解以及優(yōu)化程序。下面，就和大家分享一下如何來檢測 Python 程序的運(yùn)行時(shí)間和內(nèi)存占用情況。

相關(guān)庫

Time

time 庫是 Python 處理時(shí)間的標(biāo)準(zhǔn)庫，提供了獲取系統(tǒng)時(shí)間并格式化輸出的功能，提供系統(tǒng)級精確計(jì)時(shí)功能，用于程序性能分析。

常用方法

時(shí)間獲取    time()

時(shí)間格式化    strftime(format）

程序計(jì)時(shí)    sleep(seconds)

OS

os 庫是 Python 標(biāo)準(zhǔn)庫之一，提供了通用的、基本的操作系統(tǒng)交互功能。

常用方法

打開文件   os.open()

返回當(dāng)前工作目錄    os.getcwd()

獲取當(dāng)前進(jìn)程的進(jìn)程ID    os.getpid

psutil

psutil 是一個(gè)開源且擴(kuò)平臺的庫，提供了便利的函數(shù)用來獲取系統(tǒng)的信息，例如：CPU、內(nèi)存、網(wǎng)絡(luò)、磁盤等。還可以用來進(jìn)行進(jìn)程管理。

常用方法

常看CPU個(gè)數(shù)    cpu_count()

獲取當(dāng)前的進(jìn)程對象    Process()

查看網(wǎng)卡信息統(tǒng)計(jì)    net_io_counters()

代碼實(shí)例：

下面我們將編寫兩個(gè)函數(shù)，分別是用來檢測程序運(yùn)行時(shí)間和程序內(nèi)存占用的情況。同時(shí)編寫一個(gè)簡單的函數(shù)，用來測試。詳細(xì)代碼如下：

import time
import os
import psutil

def count_time(func):
    def int_time():
        start_time = time.time()
        func()
        over_time = time.time()
        total_time = over_time - start_time
        print("程序運(yùn)行了%s秒" % total_time)
    return int_time

def count_info(func):
    def float_info():
        pid = os.getpid()
        p = psutil.Process(pid)
        info_start = p.memory_full_info().uss/1024
        func()
        info_end=p.memory_full_info().uss/1024
        print("程序占用了內(nèi)存"+str(info_end-info_start)+"KB")
    return float_info

@count_time
@count_info
def main():
    a = [i for i in range(10000)]
    print(a)

if __name__ == '__main__':
    main()

運(yùn)行結(jié)果：

總結(jié)

以上就是關(guān)于如何檢測 Python 測序的運(yùn)行時(shí)間和內(nèi)存占用情況的全部內(nèi)容。