File tree 2 files changed +14
-1
lines changed
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):
619
619
self .df .select_dtypes (include = "int" )
620
620
621
621
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
+
622
635
from .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):
1387
1387
1388
1388
v = self .array .nbytes
1389
1389
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 )
1391
1391
return v
1392
1392
1393
1393
@doc (
You can’t perform that action at this time.
0 commit comments