在机器学习中遇到一幅图：

代码如下 ：

ax.scatter(dataMat[:, 0], dataMat[:, 1], \

               s = 20*np.array(labels), \

               c = np.array(labels)/3.)

x,y为dataMat的第0列跟第1列，每个点大小跟颜色由labels中元素一一决定，labels列表元素只含有1,2,3。从图也可以看出只含三类元素。想给这个图加个legend，说明三类点各代表什么。

在matplotlib官网只找到一个legend_scatter的例子

http://matplotlib.org/examples/pylab_examples/legend_scatter.html

不过似乎这儿不能这么用。请问大家有什么好方法