divide an array into contiguous k sub arrays

divide_array_sub_arrays_v1.py

def divide(data, n_column=3):
    result = []
    cur_index = 0
    n = len(data)
    while cur_index < n/n_column:
        result.append(data[cur_index*n_column: (cur_index+1)*n_column])
        cur_index += 1
    if n%n_column != 0:
        result.append(data[cur_index*n_column: (cur_index+1)*n_column])
    return result

divide_array_sub_arrays_v2.py

def divide(data, n_column=3):
    result = []
    cur_index = 0
    n = len(data)
    while cur_index <= n//n_column:
        arr = data[cur_index*n_column: (cur_index+1)*n_column]
        if arr:
        	result.append(arr)
        cur_index += 1
    return result