[Matplotlib-users] Scatter legend with colored marker

[Jorge S. <jor...@ya...>]

Hi,
I am adding several scatter plots to the same axis, each having a specific
color. When I call legend on the axis it correctly picks all scatter plots with
their symbols and labels, but it doesn't pick up the color. The example below
demonstrates this. What I would like to do is to have in the legend the marker
its color matching that of the set it represents. How would I do this? 

Thanks,

Jorge

PS: I am using matplotlib 1.0

-----------
import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt

data0 = np.random.rand(10,2)
data1 = np.random.rand(10,2)
data2 = np.random.rand(10,2)
data = [data0, data1, data2]

fig, ax = plt.subplots(1,1)
norm = mpl.colors.Normalize(0,len(data))

for i,d in enumerate(data):
    ax.scatter(d.T[0], d.T[1], label='data set ' + str(i), c=np.ones(d.shape[0])*i,
            norm=norm)
ax.legend()
plt.show()
-----------