python 怎么知道一个变量存储大小

最新推荐文章于 2024-10-23 11:34:11 发布

奥莉发

最新推荐文章于 2024-10-23 11:34:11 发布

阅读量141

点赞数

python相关学习资料：

https://edu.51cto.com/video/4645.html

https://edu.51cto.com/video/3832.html

https://edu.51cto.com/video/1158.html

Python变量存储大小的探究

在Python编程中，了解变量的存储大小对于优化内存使用和程序性能至关重要。本文将探讨如何使用Python内置功能来获取变量的存储大小，并提供一个具体问题的解决方案。

问题描述

假设我们有一个大型数据集，需要存储在内存中进行处理。为了优化内存使用，我们需要知道每个变量的存储大小，以便合理分配资源。

解决方案

1. 使用`sys`模块

Python的sys模块提供了一个getsizeof函数，可以返回对象的内存大小（以字节为单位）。这是一个简单有效的方法来获取变量的存储大小。

import sys

def get_variable_size(variable):
    return sys.getsizeof(variable)

# 示例
my_list = [1, 2, 3, 4, 5]
print("Size of my_list:", get_variable_size(my_list))

2. 考虑引用类型

对于引用类型（如列表、字典等），getsizeof返回的是对象本身的大小，不包括存储在对象中的数据。因此，我们需要编写一个递归函数来计算引用类型中所有元素的总大小。

def get_total_size(obj, seen=None):
    size = sys.getsizeof(obj)
    if seen is None:
        seen = set()
    obj_id = id(obj)
    if obj_id in seen:
        return 0
    seen.add(obj_id)
    if isinstance(obj, dict):
        size += sum([get_total_size(v, seen) for v in obj.values()])
        size += sum([get_total_size(k, seen) for k in obj.keys()])
    elif hasattr(obj, '__dict__'):
        size += get_total_size(obj.__dict__, seen)
    elif hasattr(obj, '__iter__') and not isinstance(obj, (str, bytes, bytearray)):
        size += sum([get_total_size(i, seen) for i in obj])
    return size

# 示例
my_dict = {'a': 1, 'b': [1, 2, 3], 'c': {'x': 10, 'y': 20}}
print("Size of my_dict:", get_total_size(my_dict))