MATLABで 'すべての組み合わせ'をコーディングするには？

Question

Bは配列です。 B = [1 2 1]場合は、私の予想される出力は次のようになります。[0 0 0], [1 0 0], [1 1 0], [1 1 1], [0 1 1]、などと私の機能は以下の通りです：

function [output] = AllCombination(B) 
output = []; 
for i = 0:B(1) 
    for j = 0:B(2) 
     for k = 0:B(3) 
      output = [output; [i,j,k]]; 
     end 
    end 
end 

しかし、Bは、任意の長さにすることができます。 Bの長さに適応するためには、どうすればよいですか？

Answer 1

function [output] = AllCombination(B) 
C = arrayfun(@(x)0:x, B, 'UniformOutput', false); 
D = cell(1,numel(B)); 
[D{:}]=ndgrid(C{:}); 
output = cell2mat(cellfun(@(x)x(:), D, 'UniformOutput', false)); 

最初の行は、範囲を含むセル配列を構成します0から各要素のBまで第2行は、第1行の範囲を渡す3行目の出力を格納するために、正しいサイズの空のセル配列を作成してndgridにします。これにより、範囲のすべての組み合わせが構築されます。最後にcellfunを適用して、それぞれを列ベクトルに変換し、cell2matで連結します。

Answer 2

別の解決策：

L = prod(B+1); % length of the resulting array 
output = zeros(L,length(B)); % preallocating memory 
for ii = 1:length(B) % for each column of the output array 
    output(:,ii) = imresize(repmat(0:B(ii),1,prod(B(1:(ii-1))+1)), [1 L], 'nearest')'; 
end 

説明：

repmat(0:B(ii),1,prod(B(1:(ii-1))+1) 

繰り返し0:B(ii)何度でもその前にBさんの全ての要素の積としてシーケンス。カウントがゼロから始まることを考慮に入れるために、すべての要素に+1を追加します。

L = prod(B+1); % length of the resulting array 
output = zeros(L,length(B)); % preallocating memory 
for ii = 1:length(B) % for each column of the output array 
    p = prod(B(1:(ii-1))+1); 
    output(:,ii) = interp1(1:(((B(ii)+1)*p)), repmat(0:B(ii),1,p), linspace(1, (((B(ii)+1)*p)), L), 'nearest'); 
end 

：あなたは、画像処理ツールボックスを持っていない場合は interp1の代わり imresizeを使用して

バージョン：

imresize(...,[1 L], 'nearest')'; 

は、アレイ

編集の長さに、その後のベクトルをスケーリングします