使用Matplotlib中的scatter()在3D散点图中添加图例

时间:2013-12-10 21:00:43

标签: python matplotlib

我想在同一个绘图中创建一个包含不同数据集的3D散点图,并使用其标签创建一个图例。我面临的问题是我无法正确添加图例,我得到一个带有空标签的情节作为http://tinypic.com/view.php?pic=4jnm83&s=5#.Uqd-05GP-gQ中的数字。更具体地说,我得到错误: “warnings.warn(”图例不支持%s \ n使用代理艺术家。\ n \ nhttp://matplotlib.sourceforge.net/users/legend_guide.html#using-proxy-artist \ n“%(str(orig_handle) ))) /usr/lib/pymodules/python2.7/matplotlib/legend.py:610:UserWarning:Legend不支持 请改用代理艺术家。“

请在下面找到我迄今为止尝试过的示例演示:

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
import random
import csv
from os import listdir
from os.path import isfile, join

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

handles = []
colors = ['blue', 'red']

X1 = range(0,10)
Y1 = range(0,10)
Z1 = range(0,10)

random.shuffle(X1)
random.shuffle(Y1)
random.shuffle(Z1)

scatter1 = ax.scatter(X1, Y1, Z1, c = colors[0], marker = 'o')

random.shuffle(X1)
random.shuffle(Y1)
random.shuffle(Z1)

scatter2 = ax.scatter(X1, Y1, Z1, c = colors[1], marker = 'v')

ax.set_xlabel('X', fontsize = 10)
ax.set_ylabel('Y', fontsize = 10)
ax.set_zlabel('Z', fontsize = 10)

ax.legend([scatter1, scatter2], ['label1', 'label2'])

plt.show()

我见过其他大致相似的例子,但没有一个使用scatter()图。除了工作解决方案,有人可以解释我做错了什么吗?

1 个答案:

答案 0 :(得分:14)

scatter1_proxy = matplotlib.lines.Line2D([0],[0], linestyle="none", c=colors[0], marker = 'o')
scatter2_proxy = matplotlib.lines.Line2D([0],[0], linestyle="none", c=colors[1], marker = 'v')
ax.legend([scatter1_proxy, scatter2_proxy], ['label1', 'label2'], numpoints = 1)

问题是图例功能不支持3D散点返回的类型。所以你必须创建一个具有相同特征的“虚拟图”并将它们放在图例中。

numpoints = 1只能获得图例中的一个点 linestyle =“none”因此传奇中没有绘制线条