Jupyter Notebookで各変数の使用メモリチェック

概要

• 各変数とその使用メモリをまとめてpd.DataFrameとして出力

動機

• Jupyterでデータ分析しててメモリ食いまくってる変数(dataframeとか)を探してdelしたい

環境

• Python3
• Jupyter Notebook

実行内容

以下をcellに入力して実行

import sys
import pandas as pd

mem_cols = ['Variable Name', 'Memory']
memory_df = pd.DataFrame(columns=mem_cols)

for var_name in dir():
    if not var_name.startswith('_'):
        memory_df = memory_df.append(pd.DataFrame([[var_name, sys.getsizeof(eval(var_name))]], columns=mem_cols))

memory_df = memory_df.sort_values(by='Memory', ascending=False).reset_index(drop=True)
display(memory_df)