如何使用matplotlib为3D散点图创建标记化图例？

Question

我创建了一个使用颜色和标记尺寸以及独立尺寸的3D散点图。我添加了一个颜色条图例，但无法弄清楚如何为标记尺寸创建图例。

目前，情节如下：

理想情况下，我想创建一个具有6个标记，逐渐增加大小在0到250之间的图例。

例如：

Size | Label
0    | 'Small'
50   | 'Less Small'
100  | 'Getting There'
150  | 'Above Average'
200  | 'Pretty Big'
250  | 'Large'

这是生成情节的代码。输入变量xp，yp，zp，cp和co是我在别处生成的列表。

mat.style.use(['ggplot'])
fig = plt.figure(figsize=(20,15));

norm = mat.colors.Normalize(vmin=min(co), vmax=max(co))
mapper = cm.ScalarMappable(norm=norm, cmap=cm.RdBu)
cc = list()
    for v in co:
cc.append(mapper.to_rgba(v))

ax = fig.add_subplot(111, projection='3d');
ax.scatter(xp, yp, zp, s=cp, c=cc, depthshade=True, linewidth=1, edgecolor='black');

mapper.set_array(cc)
plt.colorbar( mapper, label='Count' )

y = lambda x: (np.tanh(-x*50.)+1.)*250
x = np.arange(0,.05,.01)

sizes = y(x)
labels = x

# I can't figure out what to do here:
l = ax.legend(sizes, scatterpoints = len(sizes))

Answer 1

我发现这样做的最好方法是添加一些构建图例的虚拟数据。

sizes = np.arange(0, 300, 50)
labels = ['size={0}'.format(s) for s in sizes]

points = [ax.scatter([], [], [], s=s, c='gray') for s in sizes]
plt.legend(points, labels, scatterpoints=1)

这会在绘图中添加一些空的艺术家对象，然后用于填充图例。您可以通过空对象调整颜色，透明度等，这将反映在图例中。