Эллипс вокруг данных в MATLAB
Я хотел бы воспроизвести следующий рисунок в MATLAB:
существует два класса точек с координатами X и Y. Я хотел бы окружить каждый класс эллипсом с одним параметром стандартного отклонения, который определяет, как далеко эллипс будет идти вдоль оси.
фигура была создана с помощью другого программного обеспечения, и я точно не понимаю, как он вычисляет эллипс.
вот данные, которые я использую для этого фигура. 1-й столбец-класс, 2-й-X, 3-й-Y. Я могу использовать gscatter
нарисовать себе очки.
A = [
0 0.89287 1.54987
0 0.69933 1.81970
0 0.84022 1.28598
0 0.79523 1.16012
0 0.61266 1.12835
0 0.39950 0.37942
0 0.54807 1.66173
0 0.50882 1.43175
0 0.68840 1.58589
0 0.59572 1.29311
1 1.00787 1.09905
1 1.23724 0.98834
1 1.02175 0.67245
1 0.88458 0.36003
1 0.66582 1.22097
1 1.24408 0.59735
1 1.03421 0.88595
1 1.66279 0.84183
];
gscatter(A(:,2),A(:,3),A(:,1))
кстати, здесь именно поэтому вопрос о том, как рисовать эллипс. Итак, нам просто нужно знать все параметры, чтобы нарисовать его.
обновление:
Я согласен с тем, что центр можно вычислить как средство координат X и Y. Вероятно, мне придется использовать анализ основных компонентов (PRINCOMP
) для каждого класса определить угол и форма. Все еще думаю...
3 ответов
рассмотрим код:
%# generate data
num = 50;
X = [ mvnrnd([0.5 1.5], [0.025 0.03 ; 0.03 0.16], num) ; ...
mvnrnd([1 1], [0.09 -0.01 ; -0.01 0.08], num) ];
G = [1*ones(num,1) ; 2*ones(num,1)];
gscatter(X(:,1), X(:,2), G)
axis equal, hold on
for k=1:2
%# indices of points in this group
idx = ( G == k );
%# substract mean
Mu = mean( X(idx,:) );
X0 = bsxfun(@minus, X(idx,:), Mu);
%# eigen decomposition [sorted by eigen values]
[V D] = eig( X0'*X0 ./ (sum(idx)-1) ); %#' cov(X0)
[D order] = sort(diag(D), 'descend');
D = diag(D);
V = V(:, order);
t = linspace(0,2*pi,100);
e = [cos(t) ; sin(t)]; %# unit circle
VV = V*sqrt(D); %# scale eigenvectors
e = bsxfun(@plus, VV*e, Mu'); %#' project circle back to orig space
%# plot cov and major/minor axes
plot(e(1,:), e(2,:), 'Color','k');
%#quiver(Mu(1),Mu(2), VV(1,1),VV(2,1), 'Color','k')
%#quiver(Mu(1),Mu(2), VV(1,2),VV(2,2), 'Color','k')
end
редактировать
Если вы хотите, чтобы эллипс представлял определенный уровень стандартного отклонения, правильный способ сделать это путем масштабирования ковариационной матрицы:
STD = 2; %# 2 standard deviations
conf = 2*normcdf(STD)-1; %# covers around 95% of population
scale = chi2inv(conf,2); %# inverse chi-squared with dof=#dimensions
Cov = cov(X0) * scale;
[V D] = eig(Cov);
Я бы попробовал следующий подход:
- вычислить центроид x-y для центра эллипса (x,y в связан вопрос)
- вычислить линию линейной регрессии, чтобы получить ориентацию главной оси эллипса (угол)
- вычислить стандартное отклонение по осям x и y
- перевести стандартные отклонения x-y, чтобы они были ортогональны линии подгонки (a,b)
я предположу, что в одной матрице есть только один набор точек, например
B = A(1:10,2:3);
эту процедуру можно воспроизвести для каждого набора данных.
- вычислить центр эллипсоида, который является средним из точек. Функция Матлаб:
mean
- центр данных. Функция Матлаб
bsxfun
- вычислить главную ось эллипсоида и их соответствующую величину. Функция Matlab:
eig
последовательные шаги проиллюстрированы ниже:
Center = mean(B,1);
Centered_data = bsxfun(@minus,B,Center);
[AX,MAG] = eig(Centered_data' * Centered_data);
столбцы AX содержат векторы, описывающие главную ось эллипсоида, а диагональ MAG содержит информацию об их величине. Чтобы построить эллипсоид, масштабируйте каждую главную ось квадратным корнем ее величины.
надеюсь, что это помогает.
А.