scatter with colors and legends

scatter_with_colors_and_legends.py

p_list = np.argmax(api.predict(dataset.x), axis=1)
color_master = sns.color_palette()
colors = [color_master[ptn] for ptn in p_list]
names = api.predict_names()

g = sns.JointGrid("x", "y", data=zz)
g = g.plot_joint(plt.scatter, c=colors, label=p_list)
g = g.plot_marginals(sns.distplot, kde=True)

bb = g.ax_joint.viewLim
x_size = bb.xmax-bb.xmin
y_size = bb.ymax-bb.ymin
for i, (name, c) in enumerate(zip(names, color_master)):
    g.ax_joint.text(bb.xmin+x_size*0.01, bb.ymax-(i+2)*y_size*0.05, "● "+name, color=c)