Plot Distribution of two classes given feature

Plot_Distribution.py

# Def plot distribution
def plot_distribution(data_select,data,h,limit=()):
    """
    Plot_Distribution to see how 2 classes are seperated on different features
    data_select = which feature we want to see
    data = the DataFrame
    h = column of the two classes
    limit = non mendatory, tuple to limit the x_axis
    """
    figsize =( 15, 8)
    sns.set_style("whitegrid")
    keys =  data[h].unique() 
    a = keys[0]
    b = keys[1]
    s = sns.FacetGrid(data, hue = h,aspect = 2.5, palette ={a : 'lime', b :'black'})
    s.map(sns.kdeplot, data_select, shade = True, alpha = 0.6)
    if not limit:
        s.set(xlim=(data[data_select].min(), data[data_select].max()))
    else:
        s.set(xlim=(limit[0], limit[1]))
    s.add_legend()
    s.set_axis_labels(data_select, 'proportion')
    s.fig.suptitle(data_select)
    plt.show()