Java?Stream?流中?Collectors.toMap?的用法詳解

更新時間：2024年01月19日 14:28:37 作者：致最長的電影

這篇文章主要介紹了Stream?流中?Collectors.toMap?的用法,Collectors.toMap()方法是把List轉(zhuǎn)Map的操作，本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下

一、Collectors.toMap 方法使用技巧

Collectors.toMap() 方法是把 List 轉(zhuǎn) Map 的操作

二、代碼案例

1、demo案例

public static void main(String[] args) {
    List<Student> list = Arrays.asList(
        new Student(1, "張三", 20, "29.8"),
        new Student(2, "李四", 25, "29.5"),
        new Student(3, "趙武", 23, "30.8"),
        new Student(4, "王六", 22, "31.8")
    );
    list 打印輸出為：[
                Student(id=1, name=張三, age=20, score=29.8), 
                Student(id=2, name=李四, age=25, score=29.5), 
                Student(id=3, name=趙武, age=23, score=30.8), 
                Student(id=4, name=王六, age=22, score=31.8)
               ]
    /**
     * id 作為 map 的key，name 作為 value
     * 結(jié)果集： {1=張三, 2=李四, 3=趙武, 4=王六}
    */
    Map<Integer, String> collect = list.stream()
        .collect(Collectors.toMap(Student::getId, Student::getName));
    System.out.println(collect);
    /**
     * id 作為 map 的 key，Student 對象作為 map 的 value
     * 結(jié)果集： {1=Student(id=1, name=張三, age=20, score=29.8), 
                2=Student(id=2, name=李四, age=25, score=29.5), 
                3=Student(id=3, name=趙武, age=23, score=30.8),
                4=Student(id=4, name=王六, age=22, score=31.8)}
    */
    Map<Integer, Student> collect1 = list.stream()
        .collect(Collectors.toMap(Student::getId, v -> v));
    System.out.println(collect1);
    /**
     * id 作為 map 的 key，Student 對象作為 map 的 value
     * 結(jié)果集： {1=Student(id=1, name=張三, age=20, score=29.8),
                2=Student(id=2, name=李四, age=25, score=29.5), 
                3=Student(id=3, name=趙武, age=23, score=30.8), 
                4=Student(id=4, name=王六, age=22, score=31.8)}
     */
    Map<Integer, Student> collect2 = list.stream()
        .collect(Collectors.toMap(Student::getId, Function.identity()));
    System.out.println(collect2);
}

1.1、如果不是對象，是數(shù)組，怎么轉(zhuǎn)成map？

String typeBanner = "A=1,B=2,C=3";
String[] typeBannerArray = typeBanner.split(",");
System.out.println(Arrays.toString(typeBannerArray));  // [A=1, B=2, C=3]
Map<String, String> typeBannerMap = Arrays.stream(typeBannerArray).collect(Collectors.toMap(
                (array) -> array.split("=")[0],
                (array) -> array.split("=")[1]
        ));
System.out.println(typeBannerMap);  // {A=1, B=2, C=3}

2、當 map 上述 id 如果重復，會報主鍵重復異常，解決辦法：

按照規(guī)范來寫的話，最好所有toMap，都要將這個異常提前考慮進去，不然有時候會報重復主鍵異常，這也是正例的寫法，上面的屬于反例的寫法。

2.1、Collectors.toMap 有三個重載方法：

toMap(Function<? super T, ? extends K> keyMapper, Function<? super T, ? extends U> valueMapper);
toMap(Function<? super T, ? extends K> keyMapper, Function<? super T, ? extends U> valueMapper,
        BinaryOperator<U> mergeFunction);
toMap(Function<? super T, ? extends K> keyMapper, Function<? super T, ? extends U> valueMapper,
        BinaryOperator<U> mergeFunction, Supplier<M> mapSupplier);

參數(shù)解釋:

1. keyMapper：Key 的映射函數(shù)，Student:getId 表示選擇 Student 的 getId 作為 map 的 key 值。

2. valueMapper：Value的映射函數(shù)，F(xiàn)unction.identity() 表示選擇將原來的對象作為 Map 的value 值。

3. mergeFunction：當 Key 沖突時，調(diào)用的合并方法。(n1,n2)->n1 中，如果 n1 與 n2 的 key 值相同，選擇 n1 作為那個 key 所對應的 value 值。

4. mapSupplier：Map 構(gòu)造器，在需要返回特定的 Map 時使用。第四個參數(shù) mapSupplier 用于返回一個任意類型的 Map 實例，比如我們希望返回的 Map 是根據(jù) Key 排序的。TreeMap::new

public static void main(String[] args) {
    List<Student> list = Arrays.asList(
        new Student(1, "張三", 20, "29.8"),
        new Student(2, "李四", 25, "29.5"),
        new Student(1, "趙武", 23, "30.8"),
        new Student(4, "王六", 22, "31.8")
    );
    /**
     * id 作為 map 的key，重復 id 的 name 合并作為 value
     * 結(jié)果集： {1=張三,趙武, 2=李四, 4=王六}
    */
    Map<Integer, String> collect = list.stream()
        .collect(Collectors.toMap(Student::getId, Student::getName, (n1, n2) -> n1 +","+ n2));
    System.out.println(collect);
    /**
     * 取前面一個 Student 對象
     * 結(jié)果集： {1=Student(id=1, name=張三, age=20, score=29.8), 
                2=Student(id=2, name=李四, age=25, score=29.5), 
                4=Student(id=4, name=王六, age=22, score=31.8)}
    */
    Map<Integer, Student> collect1 = list.stream()
        .collect(Collectors.toMap(Student::getId, Function.identity(), (n1, n2) -> n1));
    System.out.println(collect1);
    /**
     * 取后面一個 Student 對象
     * 結(jié)果集： {1=Student(id=1, name=趙武, age=23, score=30.8),
                2=Student(id=2, name=李四, age=25, score=29.5), 
                4=Student(id=4, name=王六, age=22, score=31.8)}
     */
    Map<Integer, Student> collect2 = list.stream()
        .collect(Collectors.toMap(Student::getId, Function.identity(), (n1, n2) -> n2, TreeMap::new));
    System.out.println(collect2);
}

寫案例遇到的問題有，上述第一個輸出，如果寫成 n1 + n2 ，map 第二個參數(shù)類型是對象或者是list集合，都是顯示編譯報錯狀態(tài)。

到此這篇關于Stream 流中 Collectors.toMap 的用法的文章就介紹到這了,更多相關Stream 流 Collectors.toMap用法內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: