我有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;
}
}
有更好的方法来实现这个吗?
答案 0 :(得分:5)
您当前的解决方案很好,但是使用CodePen收集器直观:
Map<Integer, List<Person>> personsMap =
persons.stream()
.collect(Collectors.groupingBy(Person::getGrade));
groupingBy
收集器的这种重载非常简单,因为它只需要一个分类器(Person::getGrade)
函数,该函数将提取密钥以对流的对象进行分组。
答案 1 :(得分:1)
如果使用Java 8,则更简单的解决方案:
Map<Integer, List<Person>> map = persons.stream()
.collect(Collectors.groupingBy(Person::getGrade));