MATLAB:基于多个值拆分矩阵

时间:2016-02-03 17:37:08

标签: matlab matrix indexing

我正在处理这种格式的矩阵:

M =
1    1    3
1    1    1
1    2    2
1    2    1
1    2    2
2    1    5
2    1    1
2    2    3
2    2    4
2    2    2
...

我想要做的是提取子矩阵,其中第一列和第二列中的值可以分组,以便:

M1 =
1    1    3
1    1    1

M2 =
1    2    2
1    2    1
1    2    2

M3 =
2    1    5
2    1    1

...

我一直在努力思考如何为此索引矩阵,我有一个可用的矩阵:

I =
1    1
1    2
2    1
2    2
...

我可以用来编制索引。我想知道我是否可以使用它,但我不是100%肯定如何。我不想使用for循环,因为矩阵可能相当大,并且复杂度的顺序可能变得非常大。

感谢您阅读!

1 个答案:

答案 0 :(得分:12)

使用uniqueaccumarray

轻松完成此操作
M = [ 1    1    3
      1    1    1
      1    2    2
      1    2    1
      1    2    2
      2    1    5
      2    1    1
      2    2    3
      2    2    4
      2    2    2 ]; %// data
[~, ~, u] = unique(M(:,1:2), 'rows'); %// unique labels of rows based on columns 1 and 2
M_split = accumarray(u(:), (1:size(M,1)).', [], @(x){M(sort(x),:)}); %'// group rows 
                                                                     % // based on labels

这给出了包含部分矩阵的cell array。在您的示例中,

M_split{1} =
     1     1     3
     1     1     1
M_split{2} =
     1     2     2
     1     2     1
     1     2     2
M_split{3} =
     2     1     5
     2     1     1
M_split{4} =
     2     2     3
     2     2     4
     2     2     2