可以使用Python的内置模块sys
来查看内存占用情况。
sys.getsizeof()
函数来获取对象占用的内存大小。例如:import sys
my_var = "Hello World"
print(sys.getsizeof(my_var)) # 输出:53
sys.getsizeof()
函数结合递归函数来获取复杂对象占用的内存大小。例如:import sys
def get_total_size(obj):
size = sys.getsizeof(obj)
if isinstance(obj, (list, tuple, set, frozenset, dict)):
size += sum(get_total_size(x) for x in obj)
elif isinstance(obj, object):
size += sum(get_total_size(x) for x in obj.__dict__.values())
return size
my_list = [1, 2, [3, 4, [5, 6]], 7, 8]
print(get_total_size(my_list)) # 输出:428
tracemalloc
模块来跟踪内存分配情况。例如:import tracemalloc
tracemalloc.start()
# 你的代码
snapshot = tracemalloc.take_snapshot()
top_stats = snapshot.statistics('lineno')
# 打印前 10 个内存占用最高的代码行
for stat in top_stats[:10]:
print(stat)
这样可以获得当前内存中最占用内存的代码行信息。
注意:tracemalloc
模块需要在Python 3.4及以上版本中使用。