Defining a function (including vector dot product) for all the points in 3D

2 次查看（过去 30 天）
AP
AP 2014-6-3
评论： Matt J 2014-6-3
I am trying to build the following function in a three dimensional domain.
where k is a constant vector, X is the position vector, c is a constant number, and t is time.
k is a vector of size [1 3], X is an array of size [NX*NY*NZ 3] that represents the points in the three-dimensional domain, c is a constant, and t is an array of size [1 NT].
The following is the setup of the problem.
dx = 0.1;
dy = 0.5;
dz = 0.1;
[x, y, z] = meshgrid( (1:100)*dx, (1:100)*dy, (1:100)*dz );
X = [x(:) y(:) z(:)];
k = [1 2 3];
c = 0.5;
t = 0:0.1:1;
I thought about using arrayfun and repeating the vector k using repmat and dot it with X in the second dimension but I don't know what I should do for the multiplication of c and t.
In fact, the following loop works but it is very slow (takes 200 seconds on my machine).
f = zeros(numel(X)/3, numel(t));
for n = 1:numel(t)
for i = 1:numel(X)/3
f(i, n) = tan(dot(k, X(i,:)+c*t(n)));
end
end
What would be an efficient way of defining the function for all the points and all the times? The output of this function, for example, looks like an array of size [NX*NY*NZ NT].

请先登录，再回答此问题。

采纳的回答

Matt J
Matt J 2014-6-3
f = tan( bsxfun(@plus, X*k(:), c*t) );
  3 个评论
显示 1更早的评论
Matt J
Matt J 2014-6-3
编辑：Matt J 2014-6-3
X*k(:) computes all the dot(X(i,:),k)'s
For column vector u and row vector v,
Y=bsxfun(@plus, u,v)
computes a matrix Y with Y(i,j)=u(i)+v(j). See "doc bsxfun".
Matt J
Matt J 2014-6-3
This is probably a bit faster
k = [1 2 3];
c = 0.5;
t = 0:0.1:1;
[x, y, z] = meshgrid( (1:100)*(k(1)*dx),
(1:100)*(k(2)*dy),
(1:100)*(k(3)*dz);
f=tan( bsxfun(@plus, x(:)+y(:)+z(:), ct) );

请先登录，再进行评论。

更多回答（1 个）

George Papazafeiropoulos
编辑：George Papazafeiropoulos 2014-6-3
% data
dx = 0.1;
dy = 0.5;
dz = 0.1;
[x, y, z] = meshgrid( (1:100)*dx, (1:100)*dy, (1:100)*dz );
X = [x(:) y(:) z(:)];
k = [1 2 3];
c = 0.5;
t = 0:0.1:1;
lt=length(t);
% engine
u=numel(X)/3;
t=t(ones(u,1),:);
X=repmat(X,lt,1);
t=t(:);
t=t(:,ones(1,3));
u1=sum(k(ones(numel(X)/3,1),:).*(X+c*t),2);
ff=tan(u1);
% result
ff=reshape(ff,u,[])

类别

在 Help Center 和 File Exchange 中查找有关 Matrices and Arrays 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by