在Matplotlib中绘制3D散点图

Question

在Matplotlib中绘制3D散点图

pythonnumpymatplotlibscipydata-visualization

13

我有一组Nx3的矩阵，使用scipy/numpy库。我想画出一个三维散点图，其中X和Y轴由矩阵的第一列和第二列的值决定，每个柱子的高度是矩阵中的第三列，柱子的数量由N决定。

每个矩阵表示不同的数据组，我希望它们每个都用不同的颜色进行绘制，并为整个图设置一个图例。

我有以下代码：

fig = pylab.figure()
s = plt.subplot(1, 1, 1)
colors = ['k', "#B3C95A", 'b', '#63B8FF', 'g', "#FF3300",
          'r', 'k']
ax = Axes3D(fig)
plots = []
index = 0

for data, curr_color in zip(datasets, colors):
    p = ax.scatter(log2(data[:, 0]), log2(data[:, 1]),
                   log2(data[:, 2]), c=curr_color, label=my_labels[index])

    s.legend()
    index += 1

    plots.append(p)

    ax.set_zlim3d([-1, 9])
    ax.set_ylim3d([-1, 9])
    ax.set_xlim3d([-1, 9])

问题在于ax.scatter函数会给绘制的图形添加透明度，我想要去掉它。此外，我还想设置xticks、yticks和zticks，你能告诉我如何做吗？

最后，尽管我为每个scatter调用都添加了label=""，但legend没有出现。请问如何让图例显示出来？

非常感谢您的帮助。

- user248237

3个回答

4

据我所知，3D散点图的图例不受直接支持。请参见此处： http://matplotlib.sourceforge.net/users/legend_guide.html#plotting-guide-legend 然而，您可以使用“代理艺术家”的技巧/解决方法，例如：

p = Rectangle((0, 0), 1, 1, fc="r")
axis.legend([p], ["Red Rectangle"])

因此，代理艺术家不会被添加到轴上，但您可以使用它来创建图例。

- mkocubinski

0

将 kwarg depthshade=False 设置为 False，问题得到了解决：

ax.scatter(np.log2(data[:, 0]), np.log2(data[:, 1]), 
               np.log2(data[:, 2]), 'o', c=curr_color,  label=my_labels[index], depthshade=False)

- mivkov

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Daan · Accepted Answer

尝试将“ax.scatter”替换为“ax.plot”，可能需要使用“o”参数以获得类似的圆。这样可以修复透明度和图例问题。

import matplotlib as mpl
from mpl_toolkits.mplot3d import Axes3D
import numpy as np
import matplotlib.pyplot as plt
from numpy.random import random

mpl.rcParams['legend.fontsize'] = 10

fig = plt.figure(1)
fig.clf()
ax = Axes3D(fig)
datasets = random((8,100,3))*512
my_labels = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']

colors = ['k', "#B3C95A", 'b', '#63B8FF', 'g', "#FF3300",
          'r', 'k']
index = 0
for data, curr_color in zip(datasets, colors):
    ax.plot(np.log2(data[:, 0]), np.log2(data[:, 1]), 
                   np.log2(data[:, 2]), 'o', c=curr_color, label=my_labels[index])
    index += 1

ax.set_zlim3d([-1, 9])
ax.set_ylim3d([-1, 9])
ax.set_xlim3d([-1, 9])

ax.set_xticks(range(0,11))
ax.set_yticks([1,2,8])
ax.set_zticks(np.arange(0,9,.5))

ax.legend(loc = 'upper left')
    
plt.draw()

plt.show()

我添加了一些代码并进行了一些调整，以获取一些样本数据并让你的演示其余部分正常工作。我认为你可以让它正常工作。

根据这里所述，设置刻度需要 August 2010 版的 mplot3d 更新。我从Sourceforge获取了最新的 mplot3d。由于我仍在运行 Python 2.6 和 Matplotlib 1.0.0，我不太确定 Matplotlib 1.0.1 是否包含此最新更新。

编辑

一个快速而简单的虚拟图表，用于实现散点图的3D透明效果：

index = 0
for data, curr_color in zip(datasets, colors):
    ax.scatter(np.log2(data[:, 0]), np.log2(data[:, 1]), 
                   np.log2(data[:, 2]), 'o', c=curr_color, label=my_labels[index])
    ax.plot([], [], 'o', c = curr_color, label=my_labels[index])                    
    index += 1