MATLAB Answers

the cyclist
0

Simplest way to add a legend to the groups of a scatter plot

Asked by the cyclist
on 4 Feb 2013
Latest activity Commented on by Anton
on 9 Apr 2014

Here is a trivial scatter plot that uses the default marker size (S=[]) and a vector C to assign three colors according to the current colormap.

M = magic(3);
S = [];
C = [1;1;1;2;2;2;3;3;3];
figure
scatter(M(:),sort(M(:)),S,C,'filled')

Is there a simple way to create a legend to the three colors?

  0 Comments

Log in to comment.

Products

2 Answers

Answer by bym
on 5 Feb 2013
Edited by bym
on 5 Feb 2013
 Accepted Answer

gscatter(M(:),sort(M(:)),C)

  1 Comment

Yet another MATLAB function that I knew and then forgot. Thanks for the reminder!

Log in to comment.


Answer by ChristianW
on 5 Feb 2013

Without gscatter fcn:

[~, I] = unique(C);
p = findobj(gca,'Type','Patch');
legend(p(I),'X','Y','Z')

  1 Comment

I'd like to add a correction to this solution, since it doesn't work properly in my case (8.2.0.701 (R2013b)). The problem appears, when the number of elements corresponding to different colors is not the same, for instance:

C = [1;1;1;1;2;3;3;3;3];

As far as I understand, order of handles provided by findobj is inverted: point corresponding to the first element of M has the last position in the array p.

I suggest the following solution:

[~, I] = unique(C); %
I = length(C) - I(length(I):-1:1); 
p = findobj(gca,'Type','Patch');
legend(p(I),'X','Y','Z');

Log in to comment.


Discover MakerZone

MATLAB and Simulink resources for Arduino, LEGO, and Raspberry Pi

Learn more

Discover what MATLAB® can do for your career.

Opportunities for recent engineering grads.

Apply Today

MATLAB Academy

New to MATLAB?

Learn MATLAB today!