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

30 lines
640 B
Python

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