1. 工具介绍

memory_profiler是Python的一个第三方库，其功能时基于函数的逐行代码分析工具。通过该库，可以对目标函数(允许分析多个函数)进行内存消耗分析，便于代码调优。

2. 安装

pip install memory_profiler

3. 使用

from memory_profiler import profile

@profile(precision=5)
def my_func():
    a = [1] * (10 ** 6)
    b = [2] * (2 * 10 ** 7)
    del b
    return a

if __name__ == "__main__":
    my_func()

参考文献

Python代码优化工具——memory_profiler-CSDN博客