distribute.py

def distribute(es,n):
  """
  >>> distribute([10,11,12],4)
  [[], [10], [11], [12]]
  >>> distribute([10,11,12,13],4)
  [[10], [11], [12], [13]]
  >>> distribute([10,11,12,13,14],4)
  [[10], [11], [12], [13, 14]]
  >>> distribute([10,11,12,13,14,15,16],4)
  [[10], [11, 12], [13, 14], [15, 16]]
  >>> distribute([10,11,12,13,14,15,16,17,18,19],4)
  [[10, 11], [12, 13, 14], [15, 16], [17, 18, 19]]
  """
  p = [ (len(es) * (i+1) / n) - (len(es) * i / n) for i in range(n) ]
  res = []
  e = enumerate(es)
  for m in p:
    res.append([next(e)[1] for i in range(m)])
  return res