Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def load_grid(path):
- with open(path) as f:
- lines = f.readlines()
- lines = list(map(str.strip, lines))
- lines = list(filter(None, lines))
- starts_with_dashes = [n for n, line in enumerate(lines) if line.startswith('--')]
- last_dash_line = starts_with_dashes[-1]
- data_lines = lines[last_dash_line + 1:]
- data = []
- for line in data_lines:
- numbers = line.split()
- for number in numbers:
- if number == '/':
- break
- if '*' in number:
- repeats, value = number.split('*')
- data.extend([float(value)] * int(repeats))
- continue
- data.append(float(number))
- side = int(len(data) ** 0.5)
- data = np.array(data).reshape(side, side)
- return data
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement