Skip to content

Instantly share code, notes, and snippets.

@jpiwowar
jpiwowar / show_importance.py
Created August 27, 2020 23:31
Convenience function for displaying feature importance list from DT or RF.
def show_importance(model,title,X,y,n=10,show_score=True):
'''
Display feature importance list for a decision tree or
random forest model, optionally displaying the accuracy score
model: Decision tree or random forest model
title: title text for plot
X: Independent variables
y: Target variables
n: Number of important features to show (default=10
show_score: calculate accuracy score (default=True)
@jpiwowar
jpiwowar / pretty_shape.py
Created August 27, 2020 23:30
Print a "nicer" version of the pandas.DataFrame.shape output
def pretty_shape(my_df,desc):
'''
Displays the shape of a pandas dataframe in a prettier format.
I don't know why the default display bugs me. It just does.
my_df: source dataframe
desc: Descriptive text. Go nuts.
'''
print(f'\n{desc}:')
display(pd.DataFrame(my_df.shape,
columns=['count'])\
@jpiwowar
jpiwowar / big_display.py
Created August 27, 2020 23:27
Adjust rows displayed for pandas dataframe in Jupyter notebook/iPython
def big_display(my_df,rows_temp=50):
'''
Purpose: Expand the default display for Jupyter notebook output, and
revert to previous boring values afterward
Requires: df: input dataframe/Series for display
rows: number of rows to display (default 50)
Warning: Managenent is not responsible for ugly notebook output if you
forget to set the cell output to scroll
'''
with pd.option_context("display.min_rows", rows_temp,