def pyramid(n):
    ans = []
    for i in range(1, n + 1):
        a = [1] * i
        ans.append(a)
    return ans


print(pyramid(0))
print(pyramid(1))
print(pyramid(2))
print(pyramid(5))

6158c3d68b40124bed86825b75bc88e3f5540c9c
<mask token>

def pyramid(n):
    ans = []
    for i in range(1, n + 1):
        a = [1] * i
        ans.append(a)
    return ans


<mask token>

def pyramid(n):
    ans = []
    for i in range(1, n + 1):
        a = [1] * i
        ans.append(a)
    return ans


print(pyramid(0))
print(pyramid(1))
print(pyramid(2))
print(pyramid(5))
