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))

n9のように機能しない数字を使用すると、どのような動作が予想されますか？

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

もう1つ、楽しみのために：

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を持つ基になるリストに依存しません）

Answer 6

これが私の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の計算は、次の正の根を持つ2次方程式の解に依存します。

0 = y*y + y - 2*n