使用 Java Map 聚合 List 中重复元素的数值

霞舞

发布时间：2025-08-23 16:02:20

557人浏览过

来源于php中文网

原创

使用 java map 聚合 list 中重复元素的数值

本文介绍了如何使用 Java Map 结构有效地聚合 List 中具有相同类型（Type）的元素的数值，例如金额（Amount）和数量（Quantity）。通过将 List 转换为 Map，并利用 compute 方法或 Stream API 的 toMap 操作，可以避免手动循环和比较，从而简化代码并提高效率。本教程提供了详细的代码示例，展示了如何创建、更新和合并 Map 中的数据，最终得到聚合后的结果。

在处理包含重复元素的 List 时，如果需要对特定字段进行聚合操作，使用 Map 是一种非常有效的方法。Map 允许我们以键值对的形式存储数据，其中键是唯一的，这使得我们可以方便地根据键（例如，元素的类型）来累加相关的值（例如，金额和数量）。

使用 Map 的 compute 方法

如果已经有一个 Map，并且需要向其中添加或更新元素，可以使用 compute 方法。该方法接受一个键和一个 BiFunction 作为参数。BiFunction 接受键和当前值（如果存在）作为输入，并返回新的值。如果键不存在，则当前值为 null。

以下示例展示了如何使用 compute 方法向 Map 中添加元素，并累加金额和数量：

立即学习“Java免费学习笔记（深入）”；

import java.util.HashMap;
import java.util.Map;

record Datav2(Double amount, Integer quantity) {}

public class AccumulateValues {

    public static void main(String[] args) {
        var map = new HashMap<>(Map.of("A", new Datav2(2.0, 3)));

        // add element to map equivalent to Data("A", 3.0, 3)
        map.compute("A", (k, v) -> {
            if (v == null) {
                v = new Datav2(0.0, 0);
            }
            return new Datav2(3.0, v.quantity() + 3); // Corrected: Use 3.0 instead of v.amount = 3.0
        });

        System.out.println(map); // Output: {A=Datav2[amount=3.0, quantity=6]}
    }
}

注意事项：

英特尔AI工具

英特尔AI与机器学习解决方案

下载

compute 方法中的 BiFunction 必须返回新的值，而不是直接修改当前值。
如果键不存在，BiFunction 应该返回一个新创建的对象。
上述代码已经修正，直接返回新的 Datav2 对象，避免了直接修改 v.amount 的问题。

使用 Stream API 的 toMap 方法

如果需要将一个 List 转换为 Map，可以使用 Stream API 的 toMap 方法。该方法接受三个参数：

keyMapper： 用于生成键的 Function。
valueMapper： 用于生成值的 Function。
mergeFunction： 用于合并具有相同键的两个值的 BinaryOperator。

以下示例展示了如何使用 toMap 方法将 List 转换为 Map，并累加金额和数量：

import java.util.List;
import java.util.stream.Collectors;

record Data(String type, Double amount, Integer quantity) {}

public class AccumulateValuesFromList {

    public static void main(String[] args) {
        var list = List.of(new Data("A", 2.0, 3),
                new Data("A", 3.0, 3),
                new Data("C", 2.0, 1),
                new Data("B", 10.0, 3),
                new Data("B", 2.0, 5));

        var collected = list
                .stream()
                .collect(Collectors.toMap(
                        // what will the key be
                        Data::type,
                        // what will the value be
                        data -> new Datav2(data.amount, data.quantity),
                        // how do we combine two values if they have the same key
                        (d1, d2) -> new Datav2(d1.amount() + d2.amount(), d1.quantity() + d2.quantity())
                ));
        System.out.println(collected); // Output: {A=Datav2[amount=5.0, quantity=6], B=Datav2[amount=12.0, quantity=8], C=Datav2[amount=2.0, quantity=1]}
    }
}

注意事项：