我如何使用cellfun或arrayfun代替这个循环?

时间:2013-02-04 06:01:18

标签: matlab matrix vectorization

我在Matlab中实现了以下功能。

function [x y] = cloud(a,b,phi,x0,y0,N)
phi=phi*2*pi/360;
m=ceil(5*N/pi);
x=a*(-1+2*rand(m,1));
y=b*(-1+2*rand(m,1));
f=sqrt(a^2 - b^2);
indexMatrix=zeros(m,3);
indexMatrix(:,1)=x;
indexMatrix(:,2)=y;
insidePointsMatrix=zeros(1,2);
j=1;
for i=1:m
    insidePoint=sqrt(((x(i)+f).^2) + (y(i))^2) + sqrt(((x(i)-f).^2) + (y(i))^2);
    if (insidePoint<=2*a)
        indexMatrix(i,3)=1;
        %insidePointsMatrix(j,:)=indexMatrix(i,1:2);
        if j<=N
            insidePointsMatrix(j,1:2)=indexMatrix(i,1:2);
            j=j+1;
        end
    end
end

我正在阅读arrayfuncellfun,我想知道是否有可能缩短我已经使用它实现的功能。给定2 x n矩阵A其中(u,v)是任何给定行的条目,如何返回列出每行的以下公式输出的矩阵?

sqrt((u+f).^2+v.^2) + sqrt((u-v)^2+v^2)

变量fa如上所述。我试图将一系列点变换为椭圆。

1 个答案:

答案 0 :(得分:2)

使用find应该会为您提供u,v坐标:

cond=sqrt(((x+f).^2) + y.^2) + sqrt(((x-f).^2) + y.^2) ;
ind=find(cond<=2*a);
u=x(ind);
v=y(ind);