Created
October 10, 2019 12:47
-
-
Save suhassumukh/691bb728c8e2e84d476ebdbf5f6bca77 to your computer and use it in GitHub Desktop.
Total size of python object in memory
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# adapted from http://code.activestate.com/recipes/577504-compute-memory-footprint-of-an-object-and-its-cont/ | |
# | |
from __future__ import print_function | |
from sys import getsizeof, stderr | |
from itertools import chain | |
from collections import deque | |
try: | |
from reprlib import repr | |
except ImportError: | |
pass | |
def total_size(o, handlers={}, verbose=False): | |
""" Returns the approximate memory footprint an object and all of its contents. | |
Automatically finds the contents of the following builtin containers and | |
their subclasses: tuple, list, deque, dict, set and frozenset. | |
To search other containers, add handlers to iterate over their contents: | |
handlers = {SomeContainerClass: iter, | |
OtherContainerClass: OtherContainerClass.get_elements} | |
""" | |
dict_handler = lambda d: chain.from_iterable(d.items()) | |
all_handlers = {tuple: iter, | |
list: iter, | |
deque: iter, | |
dict: dict_handler, | |
set: iter, | |
frozenset: iter, | |
} | |
all_handlers.update(handlers) # user handlers take precedence | |
seen = set() # track which object id's have already been seen | |
default_size = getsizeof(0) # estimate sizeof object without __sizeof__ | |
def sizeof(o): | |
if id(o) in seen: # do not double count the same object | |
return 0 | |
seen.add(id(o)) | |
s = getsizeof(o, default_size) | |
if verbose: | |
print(s, type(o), repr(o), file=stderr) | |
for typ, handler in all_handlers.items(): | |
if isinstance(o, typ): | |
s += sum(map(sizeof, handler(o))) | |
break | |
return s | |
return sizeof(o) | |
##### Example call ##### | |
if __name__ == '__main__': | |
d = dict(a=1, b=2, c=3, d=[4,5,6,7], e='a string of chars') | |
print(total_size(d, verbose=True)) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment