在Python中将数字三角形读入2d数组

Question

我想使用Python从文件中读取整数值的三角形到整数的二维数组.数字看起来像这样:

75

95 64

17 47 82

18 35 87 10

20 04 82 47 65

...

我到目前为止的代码如下:

f = open('input.txt', 'r')
arr = []
for i in range(0, 15):
    arr.append([])
    str = f.readline()
    a = str.split(' ')
    for tok in a:
        arr[i].append(int(tok[:2]))

print arr

我觉得这可以用更严格,更Pythonesque的方式来完成.你会怎么做？

Answer 1

arr = [[int(i) for i in line.split()] for line in open('input.txt')]