Java 8 辣么大(lambda)表达式不慌之—–(五)示例-Collectors中的统计、分组、排序等
还是先定义好后面做示例要用的数据:
List<User> listUser = new ArrayList<>();
listUser.add(new User("李白", 20, true));
listUser.add(new User("杜甫", 40, true));
listUser.add(new User("李清照", 18, false));
listUser.add(new User("李商隐", 23, true));
listUser.add(new User("杜牧", 39, true));
listUser.add(new User("苏小妹", 16, false));
这个User就是一个普通的Bean对象,有name(姓名)、age(年龄)、gender(性别)三个属性及对应的set/get方法。
summarizingInt 按int类型统计
IntSummaryStatistics summaryStatistics = listUser.stream().collect(Collectors.summarizingInt(User::getAge));
System.out.println("年龄平均值:" + summaryStatistics.getAverage()); // 年龄平均值:26.0
System.out.println("人数:" + summaryStatistics.getCount()); // 人数:6
System.out.println("年龄最大值:" + summaryStatistics.getMax()); // 年龄最大值:40
System.out.println("年龄最小值:" + summaryStatistics.getMin()); // 年龄最小值:16
System.out.println("年龄总和:" + summaryStatistics.getSum()); // 年龄总和:156
一个方法把统计相关的基本上都搞定了,美滋滋。这里是按int统计,同样他也有 summarizingLong、summarizingDouble
方法,跟这个类似。
嫌这个方法多余的话,也有单个的统计方法。
maxBy取最大/minBy取最小
方法定义如下:
public static <T> Collector<T, ?, Optional<T>>
maxBy(Comparator<? super T> comparator) {
return reducing(BinaryOperator.maxBy(comparator));
}
参数传的是一个Comparator,举例说明:
// 根据指定条件取最大值: 取年纪最大的人
Optional<User> optional = listUser.stream().collect(Collectors.maxBy(Comparator.comparing((user) -> {
return user.getAge();
})));
if (optional.isPresent()) { // 判断是否有值
User user = optional.get();
System.out.println("最大年纪的人是:" + user.getName()); // 输出==》 最大年纪的人是:杜甫
}
取最小的与取最大的是一样的道理:
// 根据指定条件取最小值: 取年纪最小的人
Optional<User> optional = listUser.stream().collect(Collectors.minBy(Comparator.comparing((user) -> {
return user.getAge();
})));
if (optional.isPresent()) { // 判断是否有值
User user = optional.get();
System.out.println("最小年纪的人是:" + user.getName()); // 输出==》 最小年纪的人是:苏小妹
}
averagingInt /averagingLong/averagingDouble取平均值
方法定义:
public static <T> Collector<T, ?, Double>
averagingInt(ToIntFunction<? super T> mapper) {
return new CollectorImpl<>(
() -> new long[2],
(a, t) -> { a[0] += mapper.applyAsInt(t); a[1]++; },
(a, b) -> { a[0] += b[0]; a[1] += b[1]; return a; },
a -> (a[1] == 0) ? 0.0d : (double) a[0] / a[1], CH_NOID);
}
传一个ToIntFunction类型的参数,这里说一下ToIntFunction。前面说过Function<T, R>的定义,T是入参的类型R是返回值的类型,与Function<T, R>相应的有IntFunction,他规定死入参类型是int,返回类型自定义;然后一个ToIntFunction他的入参类型是自定义,返回类型定死了是int。同理其他的Long类型的,Double类型的一样。
所以我们此处的averagingInt这样写:
ToIntFunction<User> mapper = (user)->{
return user.getAge();
};
Double averageAge = listUser.stream().collect(Collectors.averagingInt(mapper ));
System.out.println("平均年齡是:" + averageAge); // 输出--》 平均年齡是:26.0
或者
Double averageAge = listUser.stream().collect(Collectors.averagingInt(User::getAge));
System.out.println("平均年齡是:" + averageAge); // 输出--》 平均年齡是:26.0
counting计数
counting方法很简单,直接上例子:
int count = listUser.stream().collect(Collectors.counting()); // count其实就等于listUser.size();
此处单纯数个数没什么用,相当于listUser.size(); 。但是可以结合其他方法使用,如结合filter使用求男性数量:
Long count = listUser.stream().filter(user -> user.getGender()).collect(Collectors.counting());
System.out.println("男性个数:" + count); // 输出--》 男性个数:2
partitioningBy分隔/groupingBy分组
partitioningBy 方法的定义:
groupingBy方法的定义:
通过定义可以看到,partitioningBy的第一个参数是一个Predicate,而groupingBy的第一个参数是一个Function<T,R>,Predicate返回的是Boolean类型而Function<T,R>返回的是自己定义的类型。所以当分组的依据是Boolean类型的时候这两个方法使用效果一样,如:
// 将List中的人按性别分组
Predicate<User> predicate = (user) -> {
return user.getGender();
};
Map<Boolean, List<User>> partition = listUser.stream().collect(Collectors.partitioningBy(predicate));
Function<User, Boolean> classifier = (user) -> {
return user.getGender();
};
Map<Boolean, List<User>> groupby = listUser.stream().collect(Collectors.groupingBy(classifier));
上面2个方法得到的结果是一样的,因为从上面的2个函数predicate和classifier的定义可以看到,2个函数的实现是一样的。但是如果想要按其他类型的数据(如String)来进行分组的话 partitioningBy就要略逊一筹了,如按User的姓名来分组,partitioningBy就不好写了,groupingBy可以稍微改一下就行:
Function<User, String> classifier = (user) -> {
return user.getName();
};
Map<String, List<User>> groupby = listUser.stream().collect(Collectors.groupingBy(classifier));
sorted 排序
这个方法不是Collectors的方法,是Stream接口里面的一个方法。作用就是将Stream里面的元素排序,但是有个前提是里面的内容是Comparable的(或者说实现过Comparable接口的),否则会抛出异常。
如上的listUser,如果直接写:
List<User> sorted = listUser.stream().sorted().collect(Collectors.toList());
就会抛出Exception in thread "main" java.lang.ClassCastException: User cannot be cast to java.lang.Comparable
异常。
所以要先让User对象实现一下Comparable接口并重写compareTo方法:
@Override
public int compareTo(User user) {
return this.getAge().compareTo(user.getAge());
}
这样上面的sorted()方法就能按年龄来排序了。
或者User对象不实现一下Comparable接口,就直接写:
List<User> sorted3 = listUser.stream()
.sorted((x, y) -> (x.getAge() < y.getAge()) ? -1 : ((x.getAge() > y.getAge()) ? 1 : 0))
.collect(Collectors.toList());
道理是一样的,写法不一样罢了。