File tree Expand file tree Collapse file tree 2 files changed +14
-1
lines changed
Expand file tree Collapse file tree 2 files changed +14
-1
lines changed Original file line number Diff line number Diff line change @@ -619,4 +619,17 @@ def time_select_dtypes(self, n):
619619 self .df .select_dtypes (include = "int" )
620620
621621
622+ class MemoryUsage :
623+ def setup (self ):
624+ self .df = DataFrame (np .random .randn (100000 , 2 ), columns = list ("AB" ))
625+ self .df2 = self .df .copy ()
626+ self .df2 ["A" ] = self .df2 ["A" ].astype ("object" )
627+
628+ def time_memory_usage (self ):
629+ self .df .memory_usage (deep = True )
630+
631+ def time_memory_usage_object_dtype (self ):
632+ self .df2 .memory_usage (deep = True )
633+
634+
622635from .pandas_vb_common import setup # noqa: F401 isort:skip
Original file line number Diff line number Diff line change @@ -1387,7 +1387,7 @@ def memory_usage(self, deep=False):
13871387
13881388 v = self .array .nbytes
13891389 if deep and is_object_dtype (self ) and not PYPY :
1390- v += lib .memory_usage_of_objects (self .array )
1390+ v += lib .memory_usage_of_objects (self ._values )
13911391 return v
13921392
13931393 @doc (
You can’t perform that action at this time.
0 commit comments