Python's small tricks

内存监测

Python 中的 tracemalloc 模块可用于追踪内存分配。

最粗暴地，我们可以获取当前内存占用和峰值占用。

具体来说，需要先用 tracemalloc.start() 开始追踪。然后，可以使用 tracemalloc.get_traced_memory() 获取追踪的内存块的当前大小和峰值大小，单位是字节（Byte）。停止追踪时，使用 tracemalloc.stop()。

例如：

tracemalloc.start()
do_something()
peak_mem = tracemalloc.get_traced_memory()[1]
tracemalloc.stop()
print(f"Peak was: {peak_mem / 1024 ** 2:.2f} MB")