CateColourScatterplot_legend.py

import pandas as pd
import plotly
import plotly.graph_objs as go
from plotly.offline import *


# To initiate ploty to run offline
init_notebook_mode(connected=True)
# Import data
df = pd.read_csv('../Data/tips.csv')

color_scheme = {'Lunch':'red','Dinner':'blue','Coffee':'brown'}

meal_type_color = [color_scheme[meal] for meal in df['meal_type'].tolist()]

# Data
data = []
for meal in df['meal_type'].unique():
	df_temp = df[df['meal_type']==meal]
	data.append(go.Scatter(x=df_temp['grand_total'], y=df_temp['tips'],
						marker_color=color_scheme[meal],
						name=meal,
						mode='markers'))
# Layout
layout = {'title':{'text':'Everybody\'s Tipping Distribution', 'x':0.5}}

fig = go.Figure(data=data, layout=layout)

plotly.offline.plot(fig, filename='catedim_scatterplot.html')