1
Fork 0
internship/swash/processing/read_swash.py

28 lines
533 B
Python

import numpy as np
def read_nohead(path):
with open(path) as file:
return file.read()
def read_reshape(path, shape):
return np.asarray(read_nohead(path).split(), dtype=np.float64).reshape(
shape
)
def read_nohead_scalar(path, n):
return read_reshape(path, (-1, n))
def read_nohead_k(path, n, k):
return read_reshape(path, (-1, n, k))
def read_nohead_vect(path, n):
return read_reshape(path, (-1, n, 2))
def read_nohead_vect_k(path, n):
return read_reshape(path, (-1, n, 2, k))