python | tracemalloc模块，跟踪内存分配情况

在 Python 中，tracemalloc 模块用于跟踪内存分配情况，帮助开发者分析内存使用情况，尤其是查找内存泄漏等问题。tracemalloc.get_traced_memory() 是该模块中的一个函数，用于获取当前跟踪的内存使用情况。

函数解释

tracemalloc.get_traced_memory() 返回一个元组 (current, peak)，其中：

• current：当前跟踪的内存使用量（以字节为单位）。
• peak：跟踪期间达到的最大内存使用量（以字节为单位）。

例子

import tracemalloc

# 开始跟踪内存分配
tracemalloc.start()

# 创建一个较大的列表
data = [i for i in range(1000000)]

# 获取当前和峰值内存使用情况
current, peak = tracemalloc.get_traced_memory()

# 停止跟踪内存分配
tracemalloc.stop()

print(f"Current memory usage: {current / 1024 / 1024:.2f} MB")
print(f"Peak memory usage: {peak / 1024 / 1024:.2f} MB")

解释 current / 1024 / 1024

• current 是以字节为单位的内存使用量。
• 为了将其转换为更易读的单位（如 MB），需要进行单位转换：
  • 1 KB = 1024 字节
  • 1 MB = 1024 KB = (1024 \times 1024) 字节

因此，current / 1024 / 1024 的含义是：

• 将 current（字节）除以 (1024 \times 1024)，得到以 MB 为单位的内存使用量。

输出

运行上述代码，输出可能如下：

Current memory usage: 37.50 MB
Peak memory usage: 37.50 MB

总结

• tracemalloc.get_traced_memory() 返回当前和峰值内存使用量（以字节为单位）。
• current / 1024 / 1024 是将当前内存使用量从字节转换为 MB 的操作。
• 这种转换便于直观地理解内存使用情况，尤其是在处理大量数据时。