python - python中的列表列表？

Question

我需要一个好的函数来在 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]]

Answer 1

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

Answer 2

def foo(n):
    i = 1
    while i <= n:
        last = int(i * 1.5 + 1)
        yield range(i, last)
        i = last

list(foo(3))

n当您使用一个不起作用的数字（例如 9）时，您期望什么行为？

Answer 3

改编自 gs 的答案，但没有神秘的“1.5”。

def foo(n):
    i = c = 1
    while i <= n:
        yield range(i, i + c)
        i += c
        c += 1

list(foo(10))

Answer 4

这可能不是适合列表推导的情况，但我不在乎！

from math import ceil, sqrt, max

def tri(n):
    return n*(n+1) // 2

def irt(x):
    return int(ceil((-1 + sqrt(1 + 8*x)) / 2))

def foo(n):
    return [list(range(tri(i)+1, min(tri(i+1)+1, n+1))) for i in range(irt(n))]

Answer 5

这是我的 python 高尔夫条目：

>>> def foo(n):
...     def lower(i): return 1 + (i*(i-1)) // 2
...     def upper(i): return i + lower(i)
...     import math
...     x = (math.sqrt(1 + 8*n) - 1) // 2
...     return [list(range(lower(i), upper(i))) for i in  range(1, x+1)]
...
>>>
>>> for i in [1,3,6,10,15]:
...     print i, foo(i)
...
1 [[1]]
3 [[1], [2, 3]]
6 [[1], [2, 3], [4, 5, 6]]
10 [[1], [2, 3], [4, 5, 6], [7, 8, 9, 10]]
15 [[1], [2, 3], [4, 5, 6], [7, 8, 9, 10], [11, 12, 13, 14, 15]]
>>>

x 的计算依赖于具有正根的二次方程的解

0 = y*y + y - 2*n

Answer 6

还有一个，只是为了好玩：

def lol(n):
    entries = range(1,n+1)
    i, out = 1, []
    while len(entries) > i:
        out.append( [entries.pop(0) for x in xrange(i)] )
        i += 1
    return out + [entries]

（这不依赖于具有数字 1..n 的基础列表）