Java对集合进行分组
时间: 2023-08-17 10:33:55 浏览: 98
Java对集合进行分组可以使用`Collectors.groupingBy`方法。该方法是Java 8中Stream API提供的一个工具方法,可以将一个元素流按照某个属性或者依据某个规则进行分组。
`Collectors.groupingBy`方法有两个重载版本:
1. `groupingBy(Function<? super T, ? extends K> classifier)`:按照元素的某个属性进行分组,返回一个`Map<K, List<T>>`类型的结果,其中K为分类依据的类型,T为元素类型。
2. `groupingBy(Function<? super T, ? extends K> classifier, Collector<? super T, A, D> downstream)`:按照元素的某个属性进行分组,并对每个组内的元素进行进一步的聚合操作,返回一个`Map<K, D>`类型的结果,其中K为分类依据的类型,T为元素类型,D为聚合操作的结果类型。
例如,以下代码将一个字符串列表按照它们的长度进行分组:
```
List<String> list = Arrays.asList("apple", "banana", "cat", "dog", "elephant", "fig", "goat", "hippo");
Map<Integer, List<String>> groupMap = list.stream().collect(Collectors.groupingBy(String::length));
System.out.println(groupMap); // 输出: {3=[cat, dog, fig], 4=[goat], 5=[apple], 6=[banana, hippo], 8=[elephant]}
```
上述代码中,使用`String::length`作为分类依据,将字符串列表按照长度进行分组。得到的`groupMap`是一个`Map<Integer, List<String>>`类型的结果,其中键为字符串长度,值为长度为该值的字符串列表。
如果需要对每个组内的元素进行进一步的聚合操作,可以使用`Collectors.groupingBy`方法的第二个重载版本。例如,以下代码将一个字符串列表按照首字母进行分组,并统计每个组内的字符串数量:
```
List<String> list = Arrays.asList("apple", "banana", "cat", "dog", "elephant", "fig", "goat", "hippo");
Map<Character, Long> countMap = list.stream().collect(Collectors.groupingBy(s -> s.charAt(0), Collectors.counting()));
System.out.println(countMap); // 输出: {a=1, b=1, c=1, d=1, e=1, f=1, g=1, h=1}
```
上述代码中,使用`s -> s.charAt(0)`作为分类依据,将字符串列表按照首字母进行分组。对于每个分组,使用`Collectors.counting()`方法对组内的元素进行计数,得到该组内元素的数量。得到的`countMap`是一个`Map<Character, Long>`类型的结果,其中键为字符串首字母,值为该首字母开头的字符串数量。
阅读全文