# 顯示當(dāng)前 python 程序占用的內(nèi)存大小 import osimport psutil as psutildef show_memory_info(hint):# 獲取當(dāng)前進(jìn)程的進(jìn)程號pid = os.getpid()# psutil 是一個獲取系統(tǒng)信息的庫p = psutil.Process(pid)info = p.memory_full_info()memory = info.uss/1024./1024print(f"{hint} memory used: {memory} MB ")def test_iterator():# 針對迭代器的占用測試# 對于迭代器 每個元素在生成之后就會保存在內(nèi)存中# 即使占用了巨量的內(nèi)存# 內(nèi)存不夠就會出現(xiàn) OOM 錯誤show_memory_info("initing iterator")list_1 = [i for i in range(1000000)]show_memory_info("after iterator")# 不過我們并不需要在內(nèi)存中同時保存這么多的東西# 例如對元素進(jìn)行求和 我們只需要知道每個元素在相加的那一刻是多少就行了 用完就可以扔掉了print(sum(list_1))show_memory_info("after sum ")def test_generator():# 針對生成器的占用測試show_memory_info("initing generator")# 生成器的概念應(yīng)運而生 在你調(diào)用 next 的時候 才會生成下一個變量# 生成器在初始化的時候 并不需要運行一次生成操作list_1 = (i for i in range(1000000))show_memory_info("after generator")print(sum(list_1))show_memory_info("after sum ")test_iterator()print()test_generator()

運行結(jié)果：

創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎勵來咯，堅持創(chuàng)作打卡瓜分現(xiàn)金大獎

總結(jié)

以上是生活随笔為你收集整理的通过显示当前 python 程序占用的内存大小来比较生成器和迭代器(转载)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。