将List处理成Map的通用方法

Localtrant

已于 2023-08-10 11:37:53 修改

阅读量89

点赞数 1

文章标签： list java

于 2023-08-10 11:30:11 首次发布

本文链接：https://blog.csdn.net/Localtrant/article/details/119329387

版权

public static <T, K, V> Map<K, V> toMap(List<T> list, Function<T, K> keyExtractor, Function<T, V> valueExtractor) {
      Map<K, V> map = new HashMap<>();
      for (T obj : list) {
          K key = keyExtractor.apply(obj);
          V value = valueExtractor.apply(obj);
          map.put(key, value);
      }
      return map;
}

定义集合如下：

List<Map> list = new ArrayList<>();
Map m1 = new HashMap();
m1.put("k", 1);
m1.put("v",2);

Map m2 = new HashMap();
m2.put("k", 1);
m2.put("v",5);

Map m3 = new HashMap();
m3.put("k", 2);
m3.put("v",3);

list.add(m1);
list.add(m2);
list.add(m3);

//可将List<Map>转为Map<Integer,Integer>
Map<Integer, Integer> map1 = toMap(list, map -> Integer.parseInt(map.get("k").toString()), map -> Integer.parseInt(map.get("v").toString()));

//如List中存放的是类对象，可通过T:getXXX方法传入,T是类名,例如：
//Map<Integer, Integer> map1 = toMap(list, T::getId, T::getName);

//可将List<Map>转为Map<Integer,Map>
Map<Integer, Map> k = toMap(list, map -> Integer.parseInt(map.get("k").toString()), Function.identity());

如要将List<T>转为Map<K, List<T>>，可使用如下方法：

public static <T, K, V> Map<K, List<V>> toList(List<T> list, Function<T, K> keyExtractor, Function<T, V> valueExtractor) {
        Map<K, List<V>> map = new HashMap<>();
        for (T obj : list) {
            K key = keyExtractor.apply(obj);
            V value = valueExtractor.apply(obj);
            List<V> v = map.get(key);
            if (v == null) {
                v= new ArrayList<>();
                map.put(key, v);
            }
            v.add(value);
        }
        return map;
    }

使用案例如下：

//定义集合用于测试
List<Map> list = new ArrayList<>();
Map m1 = new HashMap();
m1.put("k", 1);
m1.put("v",2);

Map m2 = new HashMap();
m2.put("k", 1);
m2.put("v",5);

Map m3 = new HashMap();
m3.put("k", 2);
m3.put("v",3);

list.add(m1);
list.add(m2);
list.add(m3);


Map<Integer, List<Map>> k1 = toList(list, map -> Integer.parseInt(map.get("k").toString()), Function.identity());

Map<Integer, List<String>> k2 = toList(list, map -> Integer.parseInt(map.get("k").toString()), map -> map.get("v").toString());


//如List中存放的是类对象，可通过T:getXXX方法传入,T是类名,例如：
//Map<Integer, Integer> map1 = toMap(list, T::getId, T::getName);