本文介绍了如何使用 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]}
}
}注意事项：
- 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]}
}
}注意事项：
- keyMapper 必须返回唯一的键。如果存在重复的键，则会抛出 IllegalStateException 异常。可以使用 mergeFunction 来处理重复的键。
- mergeFunction 必须返回合并后的值。
总结
使用 Map 可以有效地聚合 List 中重复元素的数值。compute 方法适用于向已有的 Map 中添加或更新元素，而 toMap 方法适用于将 List 转换为 Map。选择哪种方法取决于具体的需求和场景。通过合理使用 Map，可以简化代码并提高效率。
