From a1ab8ad88969e9cc9d1825a7f8cccc6aa7bb03cf Mon Sep 17 00:00:00 2001 From: Dominik Demuth Date: Sat, 30 Dec 2023 18:34:44 +0100 Subject: [PATCH] add sprinkles of number signs to text saves; closes #195 --- src/nmreval/data/points.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/src/nmreval/data/points.py b/src/nmreval/data/points.py index e4895f4..b394efa 100644 --- a/src/nmreval/data/points.py +++ b/src/nmreval/data/points.py @@ -697,17 +697,20 @@ class Points: np.savetxt(path, self.toarray(err=err), header=header, fmt='%.10e') else: with path.open('w') as f: - f.write(header) + f.write('# ' + '\n# '.join(header.split('\n'))) + f.write('\n') for i, l in enumerate(self.toarray(err=err)): if self.mask[i]: f.write('\t'.join(map(lambda _x: f'{_x:.10e}', l.tolist())) + '\n') else: - f.write('#' + '\t'.join(map(lambda _x: f'{_x:.10e}', l.tolist())) + '\n') + f.write('# ' + '\t'.join(map(lambda _x: f'{_x:.10e}', l.tolist())) + '\n') def get_state(self) -> dict: - ret_dic = {'x': self._x.tolist(), - 'y': self._y.tolist(), - 'mask': (np.where(~self.mask)[0]).tolist()} + ret_dic = { + 'x': self._x.tolist(), + 'y': self._y.tolist(), + 'mask': (np.where(~self.mask)[0]).tolist() + } if np.all(self._y_err == 0): ret_dic['y_err'] = 0.0