aboutsummaryrefslogtreecommitdiff
path: root/gn2/utility/chunks.py
blob: f6e88cbe2ecf7489d4c0e14f7917bcf2670547d9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import math


def divide_into_chunks(the_list, number_chunks):
    """Divides a list into approximately number_chunks smaller lists

    >>> divide_into_chunks([1, 2, 7, 3, 22, 8, 5, 22, 333], 3)
    [[1, 2, 7], [3, 22, 8], [5, 22, 333]]
    >>> divide_into_chunks([1, 2, 7, 3, 22, 8, 5, 22, 333], 4)
    [[1, 2, 7], [3, 22, 8], [5, 22, 333]]
    >>> divide_into_chunks([1, 2, 7, 3, 22, 8, 5, 22, 333], 5)
    [[1, 2], [7, 3], [22, 8], [5, 22], [333]]
    >>>

    """
    length = len(the_list)

    if length == 0:
        return [[]]

    if length <= number_chunks:
        number_chunks = length

    chunksize = int(math.ceil(length / number_chunks))

    chunks = []
    for counter in range(0, length, chunksize):
        chunks.append(the_list[counter:counter + chunksize])

    return chunks