Java Stream分组结果异常：无对应年龄组时未生成HashMap键

原代码的问题在于Collectors.groupingBy只会为实际存在对应元素的分组创建键值对，没有匹配元素的分组（比如"21-23"）不会出现在结果Map中。要保留所有预定义分组并将无数据的分组值设为0，可以按以下方式处理：

解决方案思路 #

1. 先创建包含所有目标分组的初始Map，每个分组默认值设为0
2. 用Stream统计出有数据的分组数量
3. 将统计结果覆盖到初始Map中，最终得到包含所有分组的结果

完整代码示例 #

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

class Person {
    private int age;
    private String name;
    private String id;
    private int status;

    public Person(int age, String name, String id, int status) {
        this.age = age;
        this.name = name;
        this.id = id;
        this.status = status;
    }

    public int getAge() {
        return age;
    }
}

public class AgeGroupStat {
    public static void main(String[] args) {
        List<Person> users = Arrays.asList(
            new Person(18, "张三", "123", 1),
            new Person(19, "李四", "124", 0),
            new Person(20, "王五", "125", 1),
            new Person(24, "孙八", "128", 0),
            new Person(27, "周九", "129", 1)
        ); 

        // 初始化所有需要的分组，默认值为0
        Map<String, Long> resultMap = new HashMap<>();
        resultMap.put("18-20", 0L);
        resultMap.put("21-23", 0L);
        resultMap.put("24-26", 0L);
        resultMap.put("else", 0L);

        // 统计有数据的分组数量
        Map<String, Long> tempGroup = users.stream()
            .collect(Collectors.groupingBy(u -> {
                    int age = u.getAge();
                    if (age >= 18 && age <= 20) { 
                        return "18-20"; 
                    } else if (age >= 21 && age <= 23) { 
                        return "21-23"; 
                    } else if (age >= 24 && age <= 26) {
                        return "24-26"; 
                    } else {
                         return "else"; 
                    }
                }), Collectors.counting());

        // 用统计结果更新初始Map，有数据的分组覆盖0值，无数据的保留0
        resultMap.putAll(tempGroup);

        // 输出验证结果
        resultMap.forEach((key, value) -> System.out.println(key + ": " + value));
    }
}

额外优化建议 #

如果分组规则需要在多处复用，可以把分组逻辑抽成独立方法，让代码更整洁：

private static String getAgeGroup(Person u) {
    int age = u.getAge();
    if (age >= 18 && age <= 20) { 
        return "18-20"; 
    } else if (age >= 21 && age <= 23) { 
        return "21-23"; 
    } else if (age >= 24 && age <= 26) {
        return "24-26"; 
    } else {
         return "else"; 
    }
}

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