Matlab中获取向量唯一元素计数及指定条件元素索引的实现

没问题，我来帮你搞定这个需求！咱们先从基础逻辑入手，再一步步调整到你要的「只处理出现次数排名前5的元素」的效果。

基础回顾：获取唯一元素及其计数 #

首先，你原来的代码已经能正确拿到唯一元素和它们的出现次数，咱们先把这部分保留：

A = [1,2,2,1,1,1,5,5,5,1,1,5];
[c, ia, ic] = unique(A); % c是唯一元素，ic是原向量元素对应c的索引
c_counts = accumarray(ic, 1); % 计算每个唯一元素的出现次数

核心修改：筛选出现次数排名前5的元素 #

接下来就是关键步骤——找出出现次数最多的前5个元素，然后获取它们在原向量中的索引：

1. 对计数降序排序，拿到排序后的索引
   咱们需要按出现次数从多到少排序，这样就能直接取前5个位置的元素：

   % 降序排序计数，sorted_idx是排序后对应c_counts的原索引
   [sorted_counts, sorted_idx] = sort(c_counts, 'descend');
   

2. 取前5个元素（兼容元素不足5的情况）
   如果你的向量里唯一元素不足5个，直接取全部就好，不用报错：

   % 取前5个（或全部，若不足5）的索引
   top5_elem_idx = sorted_idx(1:min(5, length(sorted_idx)));
   % 拿到对应的唯一元素
   top5_elements = c(top5_elem_idx);
   

3. 遍历前5元素，获取原向量中的所有索引
   最后咱们逐个遍历这些元素，找出它们在原向量里的所有位置：

   % 遍历每个前5元素，输出对应的索引
   for elem = top5_elements
       % 找到原向量中等于当前元素的所有索引
       elem_indices = find(A == elem);
       fprintf('元素%d的出现次数为%d，对应的原向量索引：%s\n', ...
               elem, c_counts(find(c == elem)), num2str(elem_indices));
   end
   

完整运行示例 #

把上面的代码整合起来，运行后针对你给的A向量，输出会是：

元素1的出现次数为6，对应的原向量索引： 1  4  5  6 10 11
元素5的出现次数为4，对应的原向量索引： 7  8  9 12
元素2的出现次数为2，对应的原向量索引： 2  3

因为你的向量里只有3个唯一元素，所以就输出全部3个，符合咱们的兼容逻辑。

小提示 #

• 如果需要把所有前5元素的索引存到一个变量里，而不是打印，可以用一个cell数组来收集：

  top5_indices = cell(length(top5_elements), 1);
  for i = 1:length(top5_elements)
      top5_indices{i} = find(A == top5_elements(i));
  end
  

  之后top5_indices里就每个元素对应一个索引数组。

内容的提问来源于stack exchange，提问作者Shew