matlab - 如何在 Matlab 中对以下代码进行矢量化？

Question

y 是 5000 x 1 包含数字 1 到 10 的向量。我可以将 y 转换为 Y(5000 x 10 矩阵)，这样

Y = zeros(5000,10);
for i = 1:5000
    Y(i,y(i))=1;
end

不使用 for 循环也能达到同样的效果吗？

Answer 1

使用 implicit expansion 的解决方案:

Y = y == 1:10;

它创建一个逻辑矩阵。如果你需要一个双矩阵，你可以这样写:

Y = double(y == 1:10);