pandas_dfs_in_memory.py

import sys
import pandas as pd
# Function to retrieve a list of Pandas DataFrames and their sizes in memory
# Call the function:
# pandas_dfs_in_memory(mydir=dir(),parent_vars=globals())

def pandas_dfs_in_memory(mydir,parent_vars):

    # get a list of all the local objects with sizes
    objects=[]
    for name,obj in parent_vars.items():
        objects.append([name,sys.getsizeof(obj)])

    # get a list of Pandas DataFrames
    alldfs=[]
    for var in mydir:
        if isinstance(parent_vars[var], pd.core.frame.DataFrame)  and var[0]!='_':
            alldfs.append(var)

    # Print the objects and memory for DataFrames
    for i in objects:
        if i[0] in alldfs:
            print(i)