plot histogram with mean and median

plot_histogram.py

def plot_histogram(df, column, b=None):
  # funtion to print histogram
  # with mean and median
  # using distplot

  # set the histogram, mean and median
  g = sns.distplot(df[column], kde=False, bins=b)
  
  plt.axvline(x=df[column].mean(),
              linewidth=3,
              color='g',
              label="mean",
              alpha=0.5)
  
  plt.axvline(x=df[column].median(),
              linewidth=3,
              color='y',
              label="median",
              alpha=0.5)

  # set title, legends and labels
  plt.xlabel("{}".format(column), size=14)
  plt.ylabel("Count", size=14)
  plt.title("Distribution of {}".format(column), size=16)
  plt.legend(["mean", "median"])

  return g