将函数应用于矩阵的每一列（向量化）

Question

在不循环遍历矩阵的每一列上应用函数的最快方法是什么？

我使用的函数是pwelch，但任何函数的概念都应该相同。 目前我正在循环我的矩阵。

 X = ones(5);
    for i = 1:5 % length of the number of columns
    result = somefunction(X(:,i))
    end

有没有办法向量化这段代码？

Answer 1

你说

任何功能的概念都应该相同

实际上并非如此。根据函数，调用它的代码可以向量化或不向量化。这取决于函数的内部编写方式。从函数外部，您无法将其向量化。矢量化是在函数内部完成的，而不是从外部完成的。

如果函数是向量化的，你只需用一个矩阵调用它，该函数就会作用于每一列。例如，sum 就是这样做的。

在pwelch的情况下，你是幸运的：根据documentation（强调添加），

Pxx = pwelch(X) 返回功率谱密度 (PSD) 估计值，Pxx，...

当X 是一个矩阵时，PSD 是 为每一列独立计算并存储在相应的 Pxx的列。

所以pwelch是矢量化函数。