找到所有集合的组合

时间:2016-04-05 16:38:08

标签: matlab combinations

假设我们有<{p>}形式的2 x N矩阵

A=| a1 a2 ... aN |
  | b1 b2 ... bN |

有2 ^ N种组合如何重新排列行。我想找到包含所有组合的矩阵B

%% N=2
B=|a1 a2|
  |a1 b2|
  |b1 a2|
  |b1 b2|
%% N=3
B=|a1 a2 a3|
  |a1 a2 b3|
  |a1 b2 a3|
  |a1 b2 b3|
  |b1 a2 a3|
  |b1 a2 b3|
  |b1 b2 a3|
  |b1 b2 b3|

这与用于学习布尔代数基础知识的表(ai = 0,bi = 1)非常相似。

问题可能会扩展为从M^N x N创建M x N矩阵。

1 个答案:

答案 0 :(得分:5)

试试这个:

A = [10 20 30; 40 50 60]; %// data matrix
[m, n] = size(A); %// m: number of rows; n: number of cols
t = dec2bin(0:2^n-1)-'0'; %// generate all possible patterns
t = bsxfun(@plus, t, 1:m:n*m-1); %// convert to linear index
result = A(t); %// index into A to get result

它给出了:

result =
    10    20    30
    10    20    60
    10    50    30
    10    50    60
    40    20    30
    40    20    60
    40    50    30
    40    50    60
@Crowley

编辑

扩展上一条评论的答案: dec2bin函数已更改为dec2base,其中 base m(以下示例中我们要为每列选择三个选项)和n列。

A = [10 20;...
     40 50;...
     70 80]; %// data matrix
[m, n] = size(A); %// m: number of rows; n: number of cols
t = dec2base(0:m^n-1,m,n)-'0'; %// generate all possible patterns
t = bsxfun(@plus, t, 1:m:n*m-1); %// convert to linear index
result = A(t) %// index into A to get result

这给出了:

result =

    10    20
    10    50
    10    80
    40    20
    40    50
    40    80
    70    20
    70    50
    70    80