使用 MatLab 匿名函数生成 MxN 矩阵

Question

假设我们有以下函数：

f = @(n) sin((0:1e-3:1) .* n * pi);
g = @(n, t) cos(n .^ 2 * pi ^2 / 2 .* t);
h = @(n) f(n) * g(n, 0);

现在，我希望能够将n 的值数组输入h 并返回n 的每个值的结果总和。

我正在努力提高效率，因此我避免了新手 for-loop 方法，即填写预先分配的矩阵并对列求和。我还尝试使用 arrayfun 并将单元格转换为矩阵然后求和，但它最终比 for 循环更慢。

有人知道我该怎么做吗？

Answer 1

事实上，由于最新版本的 MATLAB 对 JIT 编译的改进，“新手”for-loop 的竞争速度将与任何其他矢量化解决方案一样快。

% array of values of n
len = 500;
n = rand(len,1);

% preallocate matrix
X = zeros(len,1001);

% fill rows
for i=1:len
    X(i,:) = h(n(i));    % call function handle
end
out = sum(X,1);

上面的速度一样快（甚至可能更快）：

XX = cell2mat(arrayfun(h, n, 'UniformOutput',false));
out = sum(XX,1);

---

编辑：

这里它是在单个矢量化调用中直接计算而无需函数句柄的：

n = rand(len,1);

t = 0;  % or any other value
out = sum(bsxfun(@times, ...
    sin(bsxfun(@times, n, (0:1e-3:1)*pi)), ...
    cos(n.^2 * t * pi^2/2)), 1);