博主在学习python爬虫的时候,需要将一个list等分成n个list。
由于能力有限,自己也没有更好的实现方式,所以就现成的拿了别人的代码直接用。
def splist(l, s):
return [l[i:i+s] for i in range(len(l)) if i%s==0]
然而很不巧,最近在项目开发的过程当中又碰到了同样的问题,当眼这次用的是Java,所以就不能复用上述的代码了。
虽说是不同的语言,但是处理的思路还是类似的。
不过真要动手写代码起来还是比较的棘手,因为Java没有python这些丰富的函数库。
网上我也找过一些类似的方法,但都不是很满意,所以还是自己花了点时间折腾来造造这个轮子。
代码实现
废话也不多说,直接附上自己写的代码(轻喷):
/**
* 平分list成n份 数据量尽可能相等
* @param list 需要平分的list
* @param n 平分成n分
* @return
*/
public static <T> List<List<T>> splitList(List<T> list, int n) {
List<List<T>> strList = new ArrayList<>();
if (list == null) return strList;
int size = list.size();
int quotient = size / n; // 商数
int remainder = size % n; // 余数
int offset = 0; // 偏移量
int len = quotient > 0 ? n : remainder; // 循环长度
int start = 0; // 起始下标
int end = 0; // 结束下标
List<T> tempList = null;
for (int i = 0; i < len; i++) {
if (remainder != 0) {
remainder--;
offset = 1;
} else {
offset = 0;
}
end = start + quotient + offset;
tempList = list.subList(start, end);
start = end;
strList.add(tempList);
}
return strList;
}
代码用例
public static void main(String[] args) {
List<Integer> integerList = new ArrayList<>();
// 原始数据:[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]
for (int i = 0; i < 14; i++) {
integerList.add(i);
}
List<List<Integer>> splitList = splitList(integerList, 10); // 分成10等份
System.out.println(splitList);
}
以上是按10等份对原始数据进行分组,分组后输出的结果如下:
[[0, 1], [2, 3], [4, 5], [6, 7], [8], [9], [10], [11], [12], [13]]