monthly_sales = pd.merge(prediction_df[['category', 'distributorid', 'segment']], monthly_sales, on = ['category', 'distributorid'])
monthly_sales = pd.merge(prediction_df[['category', 'distributorid', 'segment']], monthly_sales, on = ['category', 'distributorid'])