Is there a function that returns a vector or array or list of values obtained by applying a function to margins of an array or matrix?

For example, assume x is a vector of length n, and y is a vector of length m. I would like to construct the matrix the (n x m) matrix M(i,j) = f(x(i),y(j)). Of course I do not want to use loops!
Btw, in R, there is such a function (called apply).

 采纳的回答

Use meshgrid()
[X, Y] = meshgrid(x,y);
M = f(X, Y);
You might need to use reshape() after that to make M be 2D
M = reshape(M, m, n);

3 个评论

Thank you very much. However, to get exactly what I wanted, you need to the transpose!
To avoid the transpose, simply reverse the initial indexing. E.g.,
[Y, X] = meshgrid(y,x);
You need to be careful. meshgrid() works on x and y, while other functions, like reshape() work on rows and columns. Note, you used x and y and said x had n columns and y had m rows. However you said that M was nxm, meaning n rows and m columns, which is the reverse. So you either made the very common mistake of thinking (rows,columns) was the same as (x,y) or you intentionally thought that your "f" function would do a transpose. Either way, you just have to be careful.

请先登录,再进行评论。

更多回答(2 个)

A bit clunky, but this works and makes no assumptions about f being vectorized:
n = numel(x);
m = numel(y);
M = arrayfun(@f,repmat(x(:),1,m),repmat(y(:)',n,1));
I'm surprised nobody mentioned this.
fun = @(x,y) sin(x+y);
x = 0:5;
y = (0:2:6)';
M = bsxfun(fun,x,y)
M =
0 0.84147 0.9093 0.14112 -0.7568 -0.95892
0.9093 0.14112 -0.7568 -0.95892 -0.27942 0.65699
-0.7568 -0.95892 -0.27942 0.65699 0.98936 0.41212
-0.27942 0.65699 0.98936 0.41212 -0.54402 -0.99999

3 个评论

Easy enough to ensure that X and Y are always the proper shapes.
M = bsxfun(fun,x(:).',y(:));
I guess I wasn't clear. bsxfun passes column vectors to the function, so whatever the function is, it must be able to deal with column vector inputs (i.e., it must be vectorized to this extent). That was the only point I was trying to make. It is similar to the f(X,Y) used in IA's solution ... f must be vectorized for this to work.

请先登录,再进行评论。

类别

帮助中心File Exchange 中查找有关 Creating and Concatenating Matrices 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by