matlab - 通过对行和列的谓词生成 bool 矩阵

Question

我有以下向量:

y = [1; 3; 2; 3; 1];

它的所有值都在 1 和 n 之间(在本例中为 3)，表示不同的选项。

我想创建一个大小为 size(y, 1) x n 的矩阵，其行对应于 y 值:

1 0 0 % because y(1) = 1
0 0 1 % because y(2) = 3
0 1 0 % because y(3) = 2
0 0 1
1 0 0

一种方法是

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

是否有更好的方法来做到这一点，也许是在一个表达式中？
基本上，我需要的是生成一个带有 bool 谓词的矩阵 (i, j) => j == y(i)。

Answer 1

如果 a 是一个列向量，你可以试试这个

a =  [1; 3; 2; 3; 1];

bsxfun(@eq, a, [1:max(a)]) 

这是一个行向量

a =  [1; 3; 2; 3; 1]';

bsxfun(@eq, a', [1:max(a)])