返回一个三角形的列表列表

Question

如何从指定的数字返回三角形列表？

例:

>>> myList = triangle(3)
             [0,0,0]
             [1,1]
             [2]
>>> myList 
[[0,0,0], [1,1], [2]]

Answer 1

试试这个:

def triangle(number):
    return [[n]*(number-n) for n in range(number)]

样本输出:

>>> triangle(4)
[[0, 0, 0, 0], [1, 1, 1], [2, 2], [3]]
>>> triangle(5)
[[0, 0, 0, 0, 0], [1, 1, 1, 1], [2, 2, 2], [3, 3], [4]]
>>> triangle(3)
[[0, 0, 0], [1, 1], [2]]

打印三角形:

>>> print '\n'.join(map(str, triangle(3)))
[0, 0, 0]
[1, 1]
[2]