diff --git a/ompy/filehandling.py b/ompy/filehandling.py index 30192b37..21cf2bd5 100644 --- a/ompy/filehandling.py +++ b/ompy/filehandling.py @@ -284,25 +284,29 @@ def load_txt_2D(path: Union[str, Path] def load_numpy_1D(path: Union[str, Path]) -> Tuple[np.ndarray, np.ndarray]: - E, values = np.load(path) + vec = np.load(path) + E = vec[:, 0] + values = vec[:, 1] return values, E def save_numpy_1D(values: np.ndarray, E: np.ndarray, path: Union[str, Path]) -> None: - mat = np.vstack((E, values)) + mat = np.column_stack((E, values)) np.save(path, mat) def load_txt_1D(path: Union[str, Path]) -> Tuple[np.ndarray, np.ndarray]: - E, values = np.loadtxt(path) + vec = np.loadtxt(path) + E = vec[:, 0] + values = vec[:, 1] return values, E def save_txt_1D(values: np.ndarray, E: np.ndarray, path: Union[str, Path], header='E[keV] values') -> None: """ E default in keV """ - mat = np.vstack((E, values)) + mat = np.column_stack([E, values]) np.savetxt(path, mat, header=header)