Java 如何按照数量分批

在实际开发中，有时候我们需要将一个大的集合或者数组按照一定的数量分成多个小的批次进行处理，这在批量处理数据的时候非常有用。下面我们将介绍几种常见的方法来实现这个功能。

方法一：使用 for 循环

首先，我们可以使用 for 循环来实现按照数量分批。我们可以定义一个批次大小（即每个批次中元素的数量），然后在循环中从原始集合或者数组中取出对应数量的元素。

public static List<List<Integer>> batchSplit(List<Integer> list, int batchSize) {
    List<List<Integer>> result = new ArrayList<>();
    int size = list.size();
    for (int i = 0; i < size; i += batchSize) {
        int end = Math.min(size, i + batchSize);
        result.add(list.subList(i, end));
    }
    return result;
}

上面的代码中，我们首先创建一个空的二维数组 result 来保存分批后的数据，在循环中使用 subList 方法从原始列表中截取对应数量的元素，并将其添加到 result 中。需要注意的是，最后一个批次可能不满足批次大小，我们使用 Math.min 方法来确保不会越界。

方法二：使用 Stream API

Java 8 引入了 Stream API，我们可以使用 Stream API 中的 collect 方法来实现按照数量分批。同样，我们可以定义一个批次大小，然后使用 Collectors.groupingBy 方法将元素按照批次进行分组。

public static List<List<Integer>> batchSplit(List<Integer> list, int batchSize) {
    return new ArrayList<>(list.stream()
            .collect(Collectors.groupingBy(
                    i -> (list.indexOf(i) + 1) / batchSize))
            .values());
}

上面的代码中，我们使用 groupingBy 方法按照 (list.indexOf(i) + 1) / batchSize 来进行分组。这里的 (list.indexOf(i) + 1) 表示元素在原始列表中的索引加一，我们再除以批次大小来得到分组的键值。最后，我们使用 values 方法获取分组后的值，并将其转换为一个新的 ArrayList。

方法三：使用 Guava 的 Iterables.partition 方法

Guava 是一个非常流行的 Java 工具库，它提供了许多实用的工具类和方法。Guava 中的 Iterables.partition 方法可以很方便地将一个集合按照指定的大小分成多个小的分区。

public static List<List<Integer>> batchSplit(List<Integer> list, int batchSize) {
    return Lists.partition(list, batchSize);
}

上面的代码中，我们直接使用 Guava 的 Lists.partition 方法来实现按照数量分批。这个方法会返回一个非常高效的分区视图，我们可以将其转换为一个新的 ArrayList。

总结

以上就是三种常见的方法来实现 Java 中按照数量分批的功能。根据具体的需求和场景选择合适的方法，可以提高代码的可读性和性能。希望本文对你有所帮助！