plot distirbution of a variable per a categorical variable

plot_distribution.py

def plot_distribution(df, target, column_values, column_name):
  # funtion to print distribution of a continuous variable
  # for categorical data

  for value in column_values:
      subset = df[df[column_name] == value]
      g = sns.kdeplot(subset[target],
                      label=value,
                      linewidth=3)

  # set title, legends and labels
  plt.ylabel("Density", size=14)
  plt.xlabel("{}".format(target), size=14)
  plt.title("Distribution of {} per {}"
            .format(target, column_name), size=16)

  return g