31 lines
640 B
Python
31 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)
|
||
|
|
)
|