Java中List常用操作比for循环更优雅的写法示例

目录

• 引言
• 简单遍历
• 筛选符合某属性条件的List集合
• 获取某属性返回新的List集合
• 获取以某属性为key，其他属性或者对应对象为value的Map集合
• 以某个属性进行分组的Map集合
• 其他情况
• 总结

引言

使用JDK1.8之后，大部分list的操作都可以使用lambda表达式去写，可以让代码更简洁，开发更迅速。以下是我在工作中常用的lambda表达式对list的常用操作，喜欢建议收藏。

以用户表为例，用户实体代码如下：

public class User {
    private Integer id; //id

    private String name; //姓名

    private Integer age; //年龄

    private Integer departId; //所属部门id
}

List<User> list = new ArrayList<>();

简单遍历

使用lambda表达式之前，如果需要遍历list时，一般使用增强for循环，代码如下：

List<User> list = new ArrayList<>();
for (User u:list) {
    System.out.println(u.toString());
}

使用lambda表达式之后，可以缩短为一行代码：

list.forEach(u-> System.out.println(u.toString()));

筛选符合某属性条件的List集合

以筛选年龄在15-17之间的用户为例，for循环写法为：

List<User> users = new ArrayList<>();
for (User u : list) {
	if (u.getAge() >= 15 && u.getAge() <= 17) {
		users.add(u);
	}
}

使用lambda表达式写法为：

List<User> users = list.stream()
                           .filter(u -> u.getAge() >= 15 && u.getAge() <= 17)
                           .collect(Collectors.toList());

获取某属性返回新的List集合

以获取id为例，项目中有时候可能会需要根据用户id的List进行查询或者批量更新操作，这时候就需要用户id的List集合，for循环写法为：

List<Integer> ids = new ArrayList<>();
for (User u:list) {
	ids.add(u.getId());
}

lambda表达式写法为：

List<User> users = list.stream()
                           .filter(u -> u.getAge() >= 15 && u.getAge() <= 17)
                           .collect(Collectors.toList());

获取以某属性为key，其他属性或者对应对象为value的Map集合

以用户id为key（有时可能需要以用户编号为key），以id对应的user作为value构建Map集合,for循环写法为：

Map<Integer,User> userMap = new HashMap<>();
for (User u:list) {
	if (!userMap.containsKey(u.getId())){
		userMap.put(u.getId(),u);
	}
}

lambda表达式写法为：

Map<Integer,User> map = list.stream()
                                     .collect(Collectors.toMap(User::getId,
                                     Function.identity(),
                                     (m1,m2)->m1));

Function.identity()返回一个输出跟输入一样的Lambda表达式对象，等价于形如t -> t形式的Lambda表达式。
(m1,m2)-> m1此处的意思是当转换map过程中如果list中有两个相同id的对象，则map中存放的是第一个对象，此处可以根据项目需要自己写。

以某个属性进行分组的Map集合

以部门id为例，有时需要根据部门分组，筛选出不同部门下的人员，如果使用for循环写法为：

Map<Integer,List<User>> departGroupMap = new HashMap<>();
for (User u:list) {
 if (departGroupMap.containsKey(u.getDepartId())){
            departGroupMap.get(u.getDepartId()).add(u);
 }else {
            List<User> users1 = new ArrayList<>();
            users1.add(u);
            departGroupMap.put(u.getDepartId(),users1);
 }
}

使用lambda表达式写法为：

Map<Integer,List<User>> departGroupMap = list.stream()
                                                                 .collect(Collectors
                                                                 .groupingBy(User::getDepartId));

其他情况

可以根据需要结合stream()进行多个操作，比如筛选出年龄在15-17岁的用户，并且根据部门进行分组分组，如果使用for循环，代码如下：

Map<Integer,List<User>> departGroupMap = new HashMap<>();
for (User u:list) {
	if (u.getAge() >= 15 && u.getAge() <= 17) {
		if (departGroupMap.containsKey(u.getDepartId())){
			departGroupMap.get(u.getDepartId()).add(u);
		}else {
			List<User> users1 = new ArrayList<>();
			users1.add(u);
			departGroupMap.put(u.getDepartId(),users1);
		}
	}
}

使用lambda表达式，代码如下：

Map<Integer,List<User>> departGroupMap = list.stream()
                                                               .filter(u->u.getAge() >= 15 && u.getAge() <= 17)
                                                               .collect(Collectors.groupingBy(User::getDepartId));

总结

上述部分是小编在工作中遇到的常用的单个List的操作，可能在项目中还会遇到更复杂的场景，可以根据需要进行多个方法的组合使用，我的感觉是使用lambda表达式代码更加简洁明了，当然各人有各人的编码习惯，不喜勿喷。

到此这篇关于Java中List常用操作比for循环更优雅的写法的文章就介绍到这了,更多相关Java中List常用操作内容请搜索我们以前的文章或继续浏览下面的相关文章希望大家以后多多支持我们！