13

次のマトリックスがあるとします。

A = randi(10, [6 3])
     7    10     3
     5     5     7
    10     5     1
     6     5    10
     4     9     1
     4    10     1

そして、2行ごとに抽出して3次元に配置したいので、結果は次のようになります。

B(:,:,1) =
     7    10     3
     5     5     7
B(:,:,2) =
    10     5     1
     6     5    10
B(:,:,3) =
     4     9     1
     4    10     1

permute / reshape / ..を使用してワンライナーとしてよりエレガントに行う方法を考えているだけです（行列のサイズとステップはパラメータでなければならないことに注意してください）

% params
step = 5;
r = 15;
c = 3;

% data
A = randi(10, [r c]);
B = zeros(step, c, r/step); % assuming step evenly divides r

% fill
counter = 1;
for i=1:step:r
    B(:,:,counter) = A(i:i+step-1, :);
    counter = counter + 1;
end
4

2 に答える 2

13

reshapeandを使用した1行のソリューションは次のpermuteとおりです。

C = 3;          % Number of columns
R = 6;          % Number of rows
newR = 2;       % New number of rows
A = randi(10, [R C]);  % 6-by-3 array of random integers
B = permute(reshape(A.', [C newR R/newR]), [2 1 3]);

もちろん、これにはnewRが に均等に分割される必要がありRます。

于 2009-09-07T20:59:41.350 に答える
2

reshapeこれはandを使用したワンライナーpermuteですが、入力配列を転置しません -

out = permute(reshape(A,newR,size(A,1)/newR,[]),[1 3 2]);

ここで、は配列出力newRの行数です 。3D

ベンチマーク

このセクションでは、この投稿で提案されているアプローチとother solution with reshape, permute & transposeパフォーマンスを比較します。データサイズは、質問にリストされているものに比例して膨らんでいます。したがって、サイズが設定され、出力が持つように分割しA、dim -3 にエントリが含まれるようにします。60000 x 3003D200 rows300

ベンチマーク コード -

%// Input
A = randi(10, [60000 300]); %// 2D matrix
newR = 200;                 %// New number of rows

%// Warm up tic/toc.
for k = 1:50000
    tic(); elapsed = toc();
end

N_iter = 5; %// Number of iterations for each approach to run with

disp('---------------------- With PERMUTE, RESHAPE & TRANSPOSE')
tic
for iter = 1:N_iter
    [R,C] = size(A);
    B = permute(reshape(A',[C newR R/newR]),[2 1 3]); %//'
end
toc, clear B R C iter

disp('---------------------- With PERMUTE & RESHAPE')
tic
for iter = 1:N_iter
    out = permute(reshape(A,newR,size(A,1)/newR,[]),[1 3 2]);
end
toc

出力 -

---------------------- With PERMUTE, RESHAPE & TRANSPOSE
Elapsed time is 2.236350 seconds.
---------------------- With PERMUTE & RESHAPE
Elapsed time is 1.049184 seconds.
于 2016-01-01T09:13:42.610 に答える