How can I use tracemalloc to measure memory usage in Python?

How can I use tracemalloc to measure memory usage in Python?

You can use the tracemalloc module to track memory allocations in your Python code. First, you need to import it and start tracing with tracemalloc.start(). After running your code, you can use tracemalloc.get_traced_memory() to get the current and peak memory usage.

Can you show me a simple example of how to implement this?

Sure! Here's a simple example:

import tracemalloc

tracemalloc.start()

# Your code here

current, peak = tracemalloc.get_traced_memory()
print(f'Current memory usage: {current / 10**6} MB; Peak was {peak / 10**6} MB')
tracemalloc.stop()

What if I want to see where the memory was allocated?

You can use tracemalloc.take_snapshot() to take a snapshot of memory allocations. Then, you can compare snapshots to see where memory was allocated. Here's how:

snapshot = tracemalloc.take_snapshot()
for stat in snapshot.statistics('lineno'):
    print(stat)