Diagonal slices of 3D array

Question

Mark Lyubarov 2020-9-25

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/599920-diagonal-slices-of-3d-array

编辑： Bruno Luong 2020-9-26

Hello,

I have a 3D matrix C of size 2N×N×N and I need to take the diagonal slices of C and construct another 2D and 1D arrays with it. Like

D(i,j)=C(i,i,j)

D(i,j)=C(i,j,j)

D(i)=C(i,i,i)

And vice versa. If I have 1D and 2D arrays, how can I add them as diagonal slices to 3D matrix?

Is there a way to do it avoiding loops?

2 个评论
显示无隐藏无

James Tursa 2020-9-25

Please show small complete examples showing input(s) and desired output(s).

Mark Lyubarov 2020-9-25

编辑：Mark Lyubarov 2020-9-25

在 MATLAB Online 中打开

yes, it goes like this. For example, let C be random.

N=10;
C=rand(2*N,N,N);
D1=zeros(2*N,N);
for i=1:2*N
    for j=1:N
        D1(i,j)=C(i,j,j);
    end
end
D1

or

D2=zeros(2*N,N)
for i=1:N
    for j=1:N
        D2(i,j)=C(i,i,j);
        D2(i+N,j)=C(i+N,i,j);
    end
end

the second is a little bit more tricky, but if I understand how to do first one, I can always split C into 2 qubic arrays and work with it.

I am new here, I don't know how to insert output, sorry. But I suppose now it is a little bit more clear.

So can I get such D1,D2 matrices without using loops?

请先登录，再进行评论。

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

Answer 1

Ameer Hamza 2020-9-25

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/599920-diagonal-slices-of-3d-array#answer_500488

编辑：Ameer Hamza 2020-9-25

在 MATLAB Online 中打开

The question is not entirely clear, so the following is my guess, but such a problem can be solved using sub2ind().

C = % 2N*N*N
rows = 1:2*N;
cols = 1:N;
pages = 1:N;
idx = sub2ind(size(C), repelem(rows, 1, N), repmat(cols, 1, 2*N), repmat(pages, 1, 2*N));
D = reshape(C(idx), N, 2*N).'

2 个评论
显示无隐藏无

Mark Lyubarov 2020-9-25

Thank you very much, it works! And do you know if it is faster than with loops?

Ameer Hamza 2020-9-26

Try using tic and toc to find the execution time in both cases. I guess the vectorized will at least be equivalent, if not faster, then the looped version.

请先登录，再进行评论。

Answer 2

Bruno Luong 2020-9-26

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/599920-diagonal-slices-of-3d-array#answer_500755

编辑：Bruno Luong 2020-9-26

在 MATLAB Online 中打开

[I,J] = ndgrid(1:N);
D1 = zeros(2*N,N);
D1(sub2ind(size(D1),I,J)) = C(sub2ind(size(C),I,J,J));
D2 = zeros(2*N,N);
D2(sub2ind(size(D2),I,J)) = C(sub2ind(size(C),I,I,J));
D2(sub2ind(size(D2),I+N,J)) = C(sub2ind(size(C),I+N,I,J));