Java8如何将list<P>转换为Map<Integer, List<P>>?

3

我有一个 List<Person> 对象,想将其转换为 Map<Integer, List<Person>>,其中映射的键表示人员属性等级。源列表中可能有多个具有相同等级的 Person 对象,在这种情况下,我希望将它们全部分组到相应等级的 List 中,并在生成的 Map 中与之对应。

到目前为止,我有以下代码:

public class PersonMain
{
   public static void main(String[] args)
   {
    Person person1 = new Person();
    person1.setName("person1");
    person1.setGrade(1);

    Person person2 = new Person();
    person2.setName("person2");
    person2.setGrade(2);

    Person person3 = new Person();
    person3.setName("person3");
    person3.setGrade(1);

    List<Person> persons = Arrays.asList(person1, person2, person3);

    Map<Integer, List<Person>> personsMap = persons.stream()
            .collect(Collectors.toMap(Person::getGrade, PersonMain::getPerson, PersonMain::combinePerson));

    System.out.println(personsMap);
}
private static List<Person> getPerson(Person p)
{
    List<Person> persons = new ArrayList<>();
    persons.add(p);
    return persons;
}
private static List<Person> combinePerson(List<Person> oldVal, List<Person> newVal)
{
        oldVal.addAll(newVal);
        return oldVal;
    }
}

有没有更好的实现方式?


可能是Java 8 List<V>转换为Map<K, V>的重复问题 - Oleksandr Pyrohov
2个回答

5

您目前的解决方案不错,但使用groupingBy收集器更加直观:

Map<Integer, List<Person>> personsMap = 
        persons.stream()
                .collect(Collectors.groupingBy(Person::getGrade));

这个 groupingBy 收集器的重载非常简单,它只需要一个分类器函数 (Person::getGrade),该函数将提取流对象的键来进行分组。


1

如果使用Java 8,则解决方案要简单得多:

Map<Integer, List<Person>> map = persons.stream()
    .collect(Collectors.groupingBy(Person::getGrade));

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接