arrays - 如何在matlab中从数组创建等价类

Question

我正在使用 matlab 并有一个数组:

a =   
2 1 5 3 2 1 2 1

您可以看到同一个值可能出现多次。我想要一个函数，它能为每个值提供一个数组，其中包含数组中该值的索引作为输出。

使用上面的示例，输出将是:

(1 5 7)
(2 6 8)
(3) 
(4)

(1 5 7) 是输入数组中 2 的索引。 1,5 和 3 也会发生同样的情况。

这可以使用for循环等来完成。我只是想知道matlab中是否有一些内置函数。

**** 编辑 ****

可能有如下两列。

2 1 5 3 2 1 2 1
3 4 3 2 4 4 3 4

在这种情况下，输出将是

(1 7)
(2 6 8)
(3)
(4)
(5)

Answer 1

使用 unique 的第三个输出获取 a 每一列的唯一标签，然后应用 accumarray具有自定义功能:

[~, ~, kk] = unique(a.', 'rows', 'stable'); %'
result = accumarray(kk, (1:numel(kk)).', [], @(x) {sort(x).'});

这适用于任意数量的行。对于您的两行示例

a = [2 1 5 3 2 1 2 1
     3 4 3 2 4 4 3 4];

结果是

result{1} =
     1     7
result{2} =
     2     6     8
result{3} =
     3
result{4} =
     4
result{5} =
     5

如果元素顺序不重要，您可以稍微简化代码:

[~, ~, kk] = unique(a.', 'rows'); %'
result = accumarray(kk, (1:numel(kk)).', [], @(x) {x.'});

这给出了

result{1} =
     2     8     6
result{2} =
     7     1
result{3} =
     5
result{4} =
     4
result{5} =
     3