I want to read a triangle of integer values from a file into a 2D array of ints using Python. The numbers would look like this:
75
95 64
17 47 82
18 35 87 10
20 04 82 47 65
...
The code I have so far is as follows:
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
I have a feeling this could be done in a tighter, more Pythonesque way. How would you do it?
arr = [[int(i) for i in line.split()] for line in open('input.txt')]
If you love us? You can donate to us via Paypal or buy me a coffee so we can maintain and grow! Thank you!
Donate Us With