Das deutsche Python-Forum

Code: Alles auswählen

input_list = range(5)
print input_list
print split(input_list, 3)

Code: Alles auswählen

[0, 1, 2, 3, 4]
[[0], [1], [2, 3, 4]]

Code: Alles auswählen

def split(list, number_of_chunks):
    chunk_size = len(list) / number_of_chunks
    remainder = len(list) % number_of_chunks
    
    temp_list = [list[(x*chunk_size):(x*chunk_size)+chunk_size] for x in range(number_of_chunks)]
    
    if remainder != 0:
        temp_list[-1] = temp_list[-1] + list[-remainder:]
    
    
    return temp_list

Code: Alles auswählen

def split(sequence, number_of_chunks):
    if not sequence:
        return []
    
    chunk_size, remainder = divmod(len(sequence), number_of_chunks)
    
    if chunk_size:
        result = [sequence[x:x+chunk_size]
                  for x in xrange(0, number_of_chunks * chunk_size, chunk_size)]
    else:
        result = [[]]
    
    if remainder:
        result[-1].extend(sequence[-remainder:])
    
    return result