java stream sum group by

原创

mob649e81673fa5 2024-08-10 06:48:23 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e81673fa5的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用Java Stream进行分组求和的实现

在Java中，Stream API为我们提供了一种高效的方式来处理集合数据。在很多场景下，我们需要对集合中的数据进行分组并求和，这时Stream就显得非常有用。本文将向您介绍如何使用Java Stream来实现分组求和的功能，适合刚入行的小白。

整体流程

我们可以将实现“java stream sum group by”的流程分为以下几个步骤：

步骤编号	步骤描述	代码示例
1	准备数据集合	List<Item> items = ...;
2	转换成Stream	items.stream();
3	按指定字段分组	.collect(Collectors.groupingBy(...));
4	对分组结果求和	.summarizingInt(Item::getValue);
5	处理结果	result.getSum();

下面我们将详细讲解每一步的实现，确保您能够理解并应用这些知识。

第一步：准备数据集合

首先，您需要有一个数据集合，假设我们有一个Item类，其结构如下：

public class Item {
    private String category; // 分类
    private int value;       // 值

    // 构造器
    public Item(String category, int value) {
        this.category = category;
        this.value = value;
    }

    // Getter方法
    public String getCategory() {
        return category;
    }

    public int getValue() {
        return value;
    }
}

接下来，我们可以创建一个List<Item>来存储我们的数据：

import java.util.ArrayList;
import java.util.List;

List<Item> items = new ArrayList<>();
items.add(new Item("A", 10));
items.add(new Item("B", 20));
items.add(new Item("A", 30));
items.add(new Item("C", 40));
items.add(new Item("B", 50));

这里我们准备了一些Item对象，以便后续操作。

第二步：转换成Stream

我们需要将这个集合转换为Stream，以便进行流式操作：

import java.util.stream.Stream;

// 转换为Stream
Stream<Item> itemStream = items.stream();

通过items.stream()，我们将List<Item>转换为一个Stream，以便后续操作。

第三步：按指定字段分组

接下来，我们可以使用Collectors.groupingBy方法按类别分组：

import java.util.Map;
import java.util.stream.Collectors;

// 按类别分组
Map<String, List<Item>> groupedItems = itemStream.collect(Collectors.groupingBy(Item::getCategory));

在这里，我们使用了Item::getCategory作为分类依据，结果将返回一个Map，其中键是类别，值是相应的Item列表。

第四步：对分组结果求和

现在，对于每个分组，我们需要对其值进行求和：

import java.util.IntSummaryStatistics;

// 求和
Map<String, IntSummaryStatistics> summedItems = groupedItems.entrySet().stream()
        .collect(Collectors.toMap(
                Map.Entry::getKey,
                e -> e.getValue().stream().collect(Collectors.summarizingInt(Item::getValue))
        ));

在这个步骤中，我们遍历了分组后的结果，对于每个类别，我们对其Item对象的值进行求和。summarizingInt方法将返回一个IntSummaryStatistics对象，其中包含了总和、平均值、最大值等信息。

第五步：处理结果

最后，我们可以从求和结果中提取出我们关心的信息，比如总和：

// 处理结果
for (Map.Entry<String, IntSummaryStatistics> entry : summedItems.entrySet()) {
    String category = entry.getKey();
    int sum = entry.getValue().getSum();
    System.out.println("Category: " + category + ", Sum: " + sum);
}

在这里，我们打印出每个类别及其对应的总和结果。

甘特图

下面是一个简单的甘特图，展示了我们执行每一步的时间安排：

gantt
    title 使用Java Stream进行分组求和的流程
    dateFormat  YYYY-MM-DD
    section 数据准备
    准备数据集合         :a1, 2023-10-01, 1d
    section 数据处理
    转换成Stream       :after a1  , 1d
    分组                :after a1  , 1d
    求和                :after a1  , 1d
    处理结果            :after a1  , 1d