Data Science, Machine Learning, Causal Inference, AI, Linear Programming.
Treatment_3}) # Select the best treatment for each customer best_treatment = cate.idxmax(axis=1) best_value = cate.max(axis=1) # Map cost for each treatment best_cost = pd.Series([cost_dict[value] for value in best_treatment]) # Create dataframe with each customers best treatment and associated