Matplotlib 如何畫散點圖的圖例？

01-05
#!/usr/bin/python2.7 # _*_ coding: utf-8 _*_

from matplotlib import pyplot as plt

from matplotlib import font_manager
import file2matrix
matrix, labels = file2matrix.file2matrix("datingTestSet.txt")

print matrix

print labels

zhfont = matplotlib.font_manager.FontProperties(fname="/usr/share/fonts/truetype/arphic/ukai.ttc")
""" 比較好看的繪製方法 """
plt.figure(figsize=(8, 5), dpi=80)

axes = plt.subplot(111)

# 將三類數據分別取出來

# x軸代表飛行的里程數

# y軸代表玩視頻遊戲的百分比

type1_x = []

type1_y = []

type2_x = []

type2_y = []

type3_x = []

type3_y = []

print "range(len(labels)):"

print range(len(labels))

for i in range(len(labels)):

    if labels[i] == 1:  # 不喜歡

        type1_x.append(matrix[i][0])

        type1_y.append(matrix[i][1])
    if labels[i] == 2:  # 魅力一般

        type2_x.append(matrix[i][0])

        type2_y.append(matrix[i][1])
    if labels[i] == 3:  # 極具魅力

        print i, "：", labels[i], ":", type(labels[i])

        type3_x.append(matrix[i][0])

        type3_y.append(matrix[i][1])
type1 = axes.scatter(type1_x, type1_y, s=20, c="red")

type2 = axes.scatter(type2_x, type2_y, s=40, c="green")

type3 = axes.scatter(type3_x, type3_y, s=50, c="blue")

# plt.scatter(matrix[:, 0], matrix[:, 1], s=20 * numpy.array(labels),

#             c=50 * numpy.array(labels), marker="o",

#             label="test")

plt.xlabel(u"每年獲取的飛行里程數", fontproperties=zhfont)

plt.ylabel(u"玩視頻遊戲所消耗的事件百分比", fontproperties=zhfont)

axes.legend((type1, type2, type3), (u"不喜歡", u"魅力一般", u"極具魅力"), loc=2, prop=zhfont)
plt.show()
http://matplotlib.org/users/screenshots.html