Split collection into sub collections in Groovy

前端 未结 3 1851
轮回少年
轮回少年 2021-01-14 05:34

I have an array containing an unknown number of items that I would like to split into separate arrays so that each separate array contains no more than 4 items. What is the

3条回答
  •  感动是毒
    2021-01-14 06:07

    Answer by tim_yates is cool, but it throws java.lang.ArrayIndexOutOfBoundsException on empty lists (for example: [].partition(4)). This can be fixed in this way:

    List.metaClass.partition = {size ->
        if (!delegate)
            return []
    
        def rslt = delegate.inject([[]]) {ret, elem ->
            (ret.last() << elem).size() >= size ? (ret << []) : ret
        }
        !rslt.last() ? rslt[0..-2] : rslt
    }
    
    assert [].partition(4) == []
    assert [1, 2, 3, 4, 5, 6].partition(4) == [[1, 2, 3, 4], [5, 6]]
    

提交回复
热议问题