indexing diagonals out of a 3d matrix

Question

Alberto Navarro 2013-7-9

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/81570-indexing-diagonals-out-of-a-3d-matrix

I thought this would be fairly simple. I thought the diag function should be able to do this task. I have a 14x14x1045 matrix Ht, t=1,...,1045 simbolizes the number of observations. I merely want to extract the main daigonals of Ht for all t and store them in a new 3d matrix Dt which is also 14x14x1045 which contains the diagonal elements of every Ht and zeros in all other positions. However, when using the function diag(Ht) I get the following error: Error using diag First input must be 2D.

Is there a function that works like diag but for 3d matrices? I have no experience programing any functions. Could someone please provide a simple loop to make a function that fulfills this task in case no such function exists already?

I´m sorry if this question is too trivial, but I haven´t been able to find a specific answer and I seem not to be able to program such a function with my expertise level...

Thanks a lot for your help!

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Matt J 2013-7-9

3
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/81570-indexing-diagonals-out-of-a-3d-matrix#answer_91275

编辑：Matt J 2013-7-9

在 MATLAB Online 中打开

D=bsxfun(@times, eye(size(H(:,:,1))), H)

2 个评论
显示无隐藏无

Matt J 2013-7-9

编辑：Matt J 2013-7-9

在 MATLAB Online 中打开

You could also use ndSparse (Available Here) to improve memory efficiency,

D=bsxfun(@times, ndSparse(speye(size(H(:,:,1)))), H);

Alberto Navarro 2013-7-9

Great! Thanks a lot for the super fast and efficient answer!

Best regards!

请先登录，再进行评论。

Answer 2

Evan 2013-7-9

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/81570-indexing-diagonals-out-of-a-3d-matrix#answer_91277

编辑：Evan 2013-7-9

在 MATLAB Online 中打开

A = rand(14,14,1045);
B = A(logical(repmat(eye(size(A(:,:,1))),[1 1 size(A,3)])))

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

Answer 3

Kiran Sagar 2013-7-9

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/81570-indexing-diagonals-out-of-a-3d-matrix#answer_91280

在 MATLAB Online 中打开

I think this would work elegantly. But I don't know if this would take up alot of time/memory for your purposes.

for i=1:1045
Dt(:,:,i)=diag(diag(Ht(:,:,i)));
end

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Alberto Navarro 2013-7-9

Memory/time wasn´t an issue. That worked fine as well. I solved it with the first answer, but with this one, I now know how to use the for loop in the command window. Thanks for the suggestion!

请先登录，再进行评论。