如何在Java中精确且一致地保留列表中指定百分比的元素？

嘿，我懂你现在用随机逐个判断的方式过滤列表，但这种方法容易因为概率波动，最终保留的元素数量和预期的百分比有偏差，尤其是列表元素不多的时候。想要精确又一致地拿到指定百分比的元素？给你几个实用的办法：

方法一：精确计算数量后随机挑选 #

最直接的思路是先算出要保留的精确元素数量，再从列表里随机挑对应数量的元素。比如列表有10个元素，要保留30%，那就是固定挑3个，不会出现2个或4个的波动。

代码示例：

import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.Collections;

public class Main {
    public static void main(String[] args) {
        List<String> list = List.of("1", "2", "3", "4", "5", "6", "7", "8", "9", "10");
        double keepPercentage = 0.3; // 要保留30%的元素
        
        // 计算需要保留的精确数量，用四舍五入处理小数
        int keepCount = (int) Math.round(list.size() * keepPercentage);
        // 兜底处理，避免数量超出列表范围
        keepCount = Math.max(0, Math.min(keepCount, list.size()));
        
        // 先复制原列表，避免修改原数据
        List<String> tempList = new ArrayList<>(list);
        // 打乱列表顺序
        Collections.shuffle(tempList, new Random());
        // 截取前keepCount个元素，就是随机挑选的结果
        List<String> filteredList = tempList.subList(0, keepCount);
        
        System.out.println("保留的元素：" + filteredList);
    }
}

这个方法的好处是数量绝对精准，如果给Random设置固定种子（比如new Random(42)），还能复现每次的筛选结果，非常适合测试场景。唯一的小缺点是筛选后的元素顺序是打乱的，如果需要保持原列表顺序，可以看下面的方法。

方法二：保留原列表顺序的精确筛选 #

如果不想打乱原列表的元素顺序，我们可以先随机生成要保留的元素索引，把这些索引排序后再提取元素，这样就能保证筛选结果的顺序和原列表一致。

代码示例：

import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class Main {
    public static void main(String[] args) {
        List<String> list = List.of("1", "2", "3", "4", "5", "6", "7", "8", "9", "10");
        double keepPercentage = 0.3;
        int keepCount = (int) Math.round(list.size() * keepPercentage);
        keepCount = Math.max(0, Math.min(keepCount, list.size()));
        
        Random random = new Random();
        // 生成所有索引并随机排序，取前keepCount个后再按原索引排序
        List<Integer> selectedIndices = IntStream.range(0, list.size())
                .boxed()
                .sorted((a, b) -> random.nextBoolean() ? -1 : 1) // 随机打乱索引
                .limit(keepCount)
                .sorted() // 把索引按升序排列，保证元素顺序和原列表一致
                .collect(Collectors.toList());
        
        // 根据索引提取元素
        List<String> filteredList = new ArrayList<>();
        for (int index : selectedIndices) {
            filteredList.add(list.get(index));
        }
        
        System.out.println("保留的元素（原顺序）：" + filteredList);
    }
}

额外说明 #

如果你其实需要的是概率性筛选（比如每个元素独立有X%的概率被保留，允许最终数量有波动），那你原来的方法没问题。但如果要求结果严格匹配指定百分比的数量，上面的两种方法会更靠谱。

内容来源于stack exchange