python之python中的列表列表

我需要一个好的函数在 python 中执行此操作。

def foo(n): 
    # do somthing 
    return list_of_lists 
 
>> foo(6) 
   [[1], 
    [2,3], 
    [4,5,6]] 
>> foot(10) 
    [[1], 
    [2,3], 
    [4,5,6] 
    [7,8,9,10]] 

请您参考如下方法:

def foo(n): 
  lol = [ [] ] 
  i = 1 
  for x in range(n): 
    if len(lol[-1]) >= i: 
      i += 1 
      lol.append([]) 
    lol[-1].append(x) 
  return lol