File size: 153 Bytes
ea7b4ce
 
 
 
 
 
1
2
3
4
5
6
def get_numeric_columns(df):
    return df.select_dtypes(include='number').columns.tolist()

def get_all_columns(df):
    return df.columns.tolist()