为 scatter3 图创建图例（Matlab）

Question

我有一个 3 维矩阵点 X（X 是一个 Nx3 矩阵），这些点属于集群。它所属的集群由 Nx1 向量 Cluster 给出（它的值类似于 1,2,3,...）。所以，我将它绘制在scatter3 上，如下所示：

scatter3(X(:,1),X(:,2),X(:,3),15,Cluster)

它工作正常，但我想为其添加一个图例，显示彩色标记和它所代表的集群。

例如，如果我有 3 个集群，我希望有这样的图例：

<blue o> - Cluster 1
<red o> - Cluster 2
<yellow o> - Cluster 3

非常感谢您的帮助！

Answer 1

建议您使用plot3，而不是使用scatter3，这将使标记更简单：

%# find out how many clusters you have
uClusters = unique(Cluster);
nClusters = length(uClusters);

%# create colormap
%# distinguishable_colormap from the File Exchange 
%# is great for distinguishing groups instead of hsv
cmap = hsv(nClusters);

%# plot, set DisplayName so that the legend shows the right label
figure,hold on
for iCluster = 1:nClusters
    clustIdx = Cluster==uClusters(iCluster);
    plot3(X(clustIdx,1),X(clustIdx,2),X(clustIdx,3),'o','MarkerSize',15,...
       'DisplayName',sprintf('Cluster %i',uClusters(iCluster)));
end

legend('show');

Answer 2

你可以使用

• legend

代码：

h = scatter3(X(:,1),X(:,2),X(:,3),15,Cluster)
hstruct = get(h);
legend(hstruct.Children, "Cluster1", "Cluster2", "Cluter3");

或

• annotation。