pandas - 从 Pandas DataFrame.info() 获取内存使用值

Question

如何获取内存使用值(显示在函数 DataFrame.info() 的输出中并分配给变量？

Answer 1

DataFrame.memory_usage().sum()
this page上有一个例子:

In [8]: df.memory_usage()
Out[8]: 
Index                 72
bool                5000
complex128         80000
datetime64[ns]     40000
float64            40000
int64              40000
object             40000
timedelta64[ns]    40000
categorical         5800
dtype: int64

# total memory usage of dataframe
In [9]: df.memory_usage().sum()
Out[9]: 290872

查看 df.info() 的源代码表明，使用 memory_usage() 是他们如何计算 df.info() 中的实际内存使用情况:

... <last few lines of def info from pandas/frame.py>
    mem_usage = self.memory_usage(index=True, deep=deep).sum()
    lines.append("memory usage: %s\n" %
                 _sizeof_fmt(mem_usage, size_qualifier))
_put_lines(buf, lines)