matlab - 有没有更好的方法来实现这个 MATLAB 代码？

Question

Y = zeros(5000,10);y 是一个 5000 x 1 的预定义向量，由 1 到 10 的数字组成；

for i= 1:size(Y,1)
    Y(i,y(i)) = 1;
end

有没有更好更简单的方法来实现这个，因为我的其余代码是矢量化的并且不包含任何 for 循环

Answer 1

你可以使用 bsxfun:

bsxfun(@eq,y,[1:10])

您可以使用 y(i)==[1:10] 创建每一行，而不是您的代码，它最终被包装在 bsxfun 中以进行矢量化。

另一个想法是索引计算:

Y((y-1).*5000+(1:5000).')=1;