Python查看变量占用内存情况

最新推荐文章于 2024-07-28 03:56:57 发布

XerCis

最新推荐文章于 2024-07-28 03:56:57 发布

阅读量2w

点赞数 11

分类专栏： Python 文章标签： python

本文链接：https://blog.csdn.net/lly1122334/article/details/104757467

版权

Python 专栏收录该内容

524 篇文章 195 订阅

订阅专栏

def show_memory(unit='KB', threshold=1):
    '''查看变量占用内存情况

    :param unit: 显示的单位，可为`B`,`KB`,`MB`,`GB`
    :param threshold: 仅显示内存数值大于等于threshold的变量
    '''
    from sys import getsizeof
    scale = {'B': 1, 'KB': 1024, 'MB': 1048576, 'GB': 1073741824}[unit]
    for i in list(globals().keys()):
        memory = eval("getsizeof({})".format(i)) // scale
        if memory >= threshold:
            print(i, memory)


if __name__ == '__main__':
    a = [i for i in range(10000)]
    show_memory()
    # a 85