Разделить массив на основе веса куска

Вычислить общую сумму массива S. Сумма каждого фрагмента должна быть около S / K.

Затем пройдитесь по массиву, вычислив текущую сумму R. Когда R+A[i+1] - S/K станет больше, чем S/K - R, закрыть текущий чанк и сделать R=0. Продолжайте со следующим фрагментом.

Вы также можете компенсировать накапливающуюся ошибку (если она возникает), сравнивая общую сумму M чанков с M * S / K

Код на скорую руку для последнего подхода (не проверено досконально)

def chunks(lst, k):
    s = sum(lst)
    sk = s / k
    #sk = max(s / k, max(lst))
    #variant from user2052436 in comments  
    idx = 0
    chunkstart = 0
    r = 0
    res = []
    for m in range(1, k):
        for idx in range(chunkstart, len(lst)):
            km = k -m
            irest = len(lst)-idx
            if((km>=irest) or (2*r+lst[idx]>2*m*sk)) and (idx>chunkstart):
                res.append(lst[chunkstart:idx])
                chunkstart = idx
                break
            r += lst[idx]
    res.append(lst[idx:len(lst)])
    return res

print(chunks([3,1,5,2,8,3,2], 3))
print(chunks([1,1,1,100], 3))

>>>[[3, 1, 5], [2, 8], [3, 2]]
   [[1, 1], [1], [100]]