我为自己编写了一个实用程序，将列表分解为给定大小的批次。我只是想知道是否已经有任何apache commons util用于此。
public static <T> List<List<T>> getBatches(List<T> collection,int batchSize){
int i = 0;
List<List<T>> batches = new ArrayList<List<T>>();
while(i<collection.size()){
int nextInc = Math.min(collection.size()-i,batchSize);
List<T> batch = collection.subList(i,i+nextInc);
batches.add(batch);
i = i + nextInc;
}
return batches;
}
请让我知道是否有任何现有的公用事业已经相同。
当前回答
您可以使用下面的代码来获得该批列表。
Iterable<List<T>> batchIds = Iterables.partition(list, batchSize);
你需要导入谷歌番石榴库来使用上面的代码。
其他回答
类似于没有流和库的OP，但更简洁:
public <T> List<List<T>> getBatches(List<T> collection, int batchSize) {
List<List<T>> batches = new ArrayList<>();
for (int i = 0; i < collection.size(); i += batchSize) {
batches.add(collection.subList(i, Math.min(i + batchSize, collection.size())));
}
return batches;
}
利用网上的各种作弊方法，我找到了这个解决方案:
int[] count = new int[1];
final int CHUNK_SIZE = 500;
Map<Integer, List<Long>> chunkedUsers = users.stream().collect( Collectors.groupingBy(
user -> {
count[0]++;
return Math.floorDiv( count[0], CHUNK_SIZE );
} )
);
我们使用count来模拟普通的集合索引。 然后，以代数商作为桶号，将集合元素分组到桶中。 最后一个映射包含作为键的桶号，作为值的桶本身。
然后，您可以轻松地对每个桶执行操作:
chunkedUsers.values().forEach( ... );
Java 8中的一行代码是:
import static java.util.function.Function.identity;
import static java.util.stream.Collectors.*;
private static <T> Collection<List<T>> partition(List<T> xs, int size) {
return IntStream.range(0, xs.size())
.boxed()
.collect(collectingAndThen(toMap(identity(), xs::get), Map::entrySet))
.stream()
.collect(groupingBy(x -> x.getKey() / size, mapping(Map.Entry::getValue, toList())))
.values();
}
在Java 9中，你可以使用带有hasNext条件的IntStream.iterate()。所以你可以把方法的代码简化成这样:
public static <T> List<List<T>> getBatches(List<T> collection, int batchSize) {
return IntStream.iterate(0, i -> i < collection.size(), i -> i + batchSize)
.mapToObj(i -> collection.subList(i, Math.min(i + batchSize, collection.size())))
.collect(Collectors.toList());
}
使用{0,1,2,3,4,5,6,7,8,9}，getbatch (numbers, 4)的结果将是:
[[0, 1, 2, 3], [4, 5, 6, 7], [8, 9]]
如果有人正在寻找Kotlin版本，这里是
list.chunked(size)
or
list.windowed(size)
曾经有一个面试问题，我在下面写了一个=D
fun <T> batch(list: List<T>, limit: Int): List<List<T>> {
val result = ArrayList<List<T>>()
var batch = ArrayList<T>()
for (i in list) {
batch.add(i)
if (batch.size == limit) {
result.add(batch)
batch = ArrayList()
}
}
if (batch.isNotEmpty()) {
result.add(batch)
}
return result
}
