Как я могу сгенерировать эту трехмерную матрицу без циклов for в MATLAB?

Question

Я бы хотел сгенерировать матрицу N × N × 3 A так, чтобы A(:,:,i) = eye(n)*i.Как я могу сделать это без использования для циклов (то есть в векторизации)?

Answer 1

Другой вариант - использовать BSXFUN , умножая единичную матрицу с массивом 1 на 1 на 3 1,2,3

>> bsxfun(@times,eye(4),permute(1:3,[3,1,2]))
ans(:,:,1) =
     1     0     0     0
     0     1     0     0
     0     0     1     0
     0     0     0     1
ans(:,:,2) =
     2     0     0     0
     0     2     0     0
     0     0     2     0
     0     0     0     2
ans(:,:,3) =
     3     0     0     0
     0     3     0     0
     0     0     3     0
     0     0     0     3

Answer 2

Один из способов сделать это - использовать функции KRON и RESHAPE :

>> N = 4;
>> A = reshape(kron(1:3,eye(N)),[N N 3])

A(:,:,1) =

     1     0     0     0
     0     1     0     0
     0     0     1     0
     0     0     0     1

A(:,:,2) =

     2     0     0     0
     0     2     0     0
     0     0     2     0
     0     0     0     2

A(:,:,3) =

     3     0     0     0
     0     3     0     0
     0     0     3     0
     0     0     0     3

Answer 3

Если у вас есть более старая версия MATLAB до появления BSXFUN, рассмотрите эту опцию (тот же ответ, что и у @ Jonas ):

N = 4; M = 3;
A = repmat(eye(N),[1 1 M]) .* repmat(permute(1:M,[3 1 2]),[N N 1])

A(:,:,1) =
     1     0     0     0
     0     1     0     0
     0     0     1     0
     0     0     0     1
A(:,:,2) =
     2     0     0     0
     0     2     0     0
     0     0     2     0
     0     0     0     2
A(:,:,3) =
     3     0     0     0
     0     3     0     0
     0     0     3     0
     0     0     0     3