java 遍历 list map_Java集合操作 遍历list并转map

序

本文主要介绍如何使用java对list进行遍历，并将list转换成map。主要涉及jdk8的stream操作和guava工具包操作集合，并涉及到guava工具包的实现说明。

需求

有一个student list集合，包含学号，身高，体重，需要遍历list，计算出BMI值，并将其转换成 学号 -> student 的map集合。

Student.java

class Student{

private int id;

private float weight;

private float height;

private float IMB;

}

计算IMB值

Private Student calIMB(Student stu){

int imb = stu.getWeight() / (stu.getHeight() * stu.getHeight());

return stu.setIMB(imb);

}

jdk8 流式操作

流式操作的代码非常简洁，一行代码即可搞定。

List studentList;

Map resultMap = studentList.stream().

map(stu -> calIMB(stu)).

collection(Collectors.toMap(stu::getId,stu -> stu,(s1,s2)-> s1));

Stream.map() 集合内的所有元素会被当成入参，运行一次入参方法。

Collectors.toMap(stu::getId,stu -> stu,(s1,s2)-> s1) 第一个参数是返回key，第二个参数是map对应的值，第三个参数为如果key重复时，选择哪个作为map的value。

guava 工具类 Lists Maps

在工作中会遇到java7或者更低版本的环境，无法使用stream流，但是for循环又太繁琐，可读性又差，这时候可以尝试使用google的guava工具包，guava提供了便捷的集合操作工具类，来看看如何实现这个需求呢。

List studentList;

studentList = Lists.transform(studentList, new Function() {

@Nullable @Override public Student apply(@Nullable Student stu) {

return calIMB(stu);

}

}); //计算IMB值

Map resultMap = Maps.uniqueIndex(studentList, new Function() {

@Nullable @Override public Integer apply(@Nullable Student input) {

return input.getId();

}

}); //list转map

利用Lists.transform对list进行封装，使每个元素获取时都会执行第二个参数自定义function。再用Maps.uniqueIndex()将list转换成map，function的返回值就是map的key。

此处需要注意的是，Lists.transform并不会执行function，返回的只是guava封装的TransformingRandomAccessList或TransformingSequentialList，在调用get或者Iterator获取元素时，才会对获取的元素执行function方法。

Lists.transform 源码

下面看一下Lists.transform做了些什么

public static List transform(

List fromList, Function super F, ? extends T> function) {

return (fromList instanceof RandomAccess)

? new TransformingRandomAccessList(fromList, function)

: new TransformingSequentialList(fromList, function);

}

先判断list是否继承RandomAccess接口，RandomAccess是一个声明式接口，继承RandomAccess代表支持非顺序的获取元素，比如ArrayList就继承该接口，ArrayList.get(i)可以直接获取到第i个元素，而链表就需要i次迭代才可获取到数据。按RandomAccess的文档说明，for循环直接get元素进行迭代要比迭代器的操作更快一些，这点令我无法理解，理论上ArrayList的迭代耗时应该基本一致才对，也不知道有什么形式的列表满足这一含义。

下面以TransformingRandomAccessList为例来看后续实现：

private static class TransformingRandomAccessList

extends AbstractList implements RandomAccess, Serializable {

final List fromList;

final Function super F, ? extends T> function;

TransformingRandomAccessList(

List fromList, Function super F, ? extends T> function) {

this.fromList = checkNotNull(fromList);

this.function = checkNotNull(function);

}

@Override public void clear() {

fromList.clear();

}

@Override public T get(int index) {

return function.apply(fromList.get(index));

}

@Override public Iterator iterator() {

return listIterator();

}

@Override public ListIterator listIterator(int index) {

return new TransformedListIterator(fromList.listIterator(index)) {

@Override

T transform(F from) {

return function.apply(from);

}

};

}

...

}

TransformingRandomAccessList的构造方法只记录了list和function，并未进行其他操作，在调用get()方法或迭代器获取元素时，会调用function.apply()方法，多次获取同一个元素，自定义function会被重复调用多次。

那么guava包内是否有方法可以类似stream流进行转换的时候就执行function获取结果，而不是获取元素时再执行的方法呢？

FluentIterable

FluentIterable 是guava集合类中常用的一个类，主要用于过滤、转换集合中的数据，它的使用更像是stream流：

List studentList;

Map resultMap = FluentIterable.from(studentList).transform(new Function() {

@Nullable @Override public Student apply(@Nullable Student input) {

return calIMB(input);

}

}).uniqueIndex(studentList, new Function() {

@Nullable @Override public Integer apply(@Nullable Student input) {

return input.getId();

}

});

和Lists、Maps工具类不同的是，每次transform操作都会对集合进行迭代，执行function，并将迭代后的Iterable通过from再次转换成FluentIterable，用以支持链式调用，看起来很像套娃，用起来也非常简洁。

小结

迭代集合进行转换操作，jdk8及以上版本用stream和lambda写起来轻松愉快。

如果是低版本的，则可以考虑使用FluentIterable，代码也非常工整。

若转换操作非常耗时，不会重复获取某个元素，并且只会使用集合的部分数据，使用Lists.transform进行转换也有助于性能提升。