dominik/nmreval
1
0
Fork 0
forked from IPKM/nmreval
Code Pull Requests Activity

add sprinkles of number signs to text saves; closes #195 (#196)

Browse Source 
Reviewed-on: IPKM/nmreval#196
This commit is contained in:
Dominik Demuth 2023-12-30 17:35:17 +00:00
commit 5ad6456b16
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/nmreval/data/points.py
View File

@ -697,7 +697,8 @@ class Points:
np.savetxt(path, self.toarray(err=err), header=header, fmt='%.10e') np.savetxt(path, self.toarray(err=err), header=header, fmt='%.10e')
else: else:
with path.open('w') as f: 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)): for i, l in enumerate(self.toarray(err=err)):
if self.mask[i]: if self.mask[i]:
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')
@ -705,9 +706,11 @@ class Points:
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: def get_state(self) -> dict:
ret_dic = {'x': self._x.tolist(), ret_dic = {
'x': self._x.tolist(),
'y': self._y.tolist(), 'y': self._y.tolist(),
'mask': (np.where(~self.mask)[0]).tolist()} 'mask': (np.where(~self.mask)[0]).tolist()
}
if np.all(self._y_err == 0): if np.all(self._y_err == 0):
ret_dic['y_err'] = 0.0 ret_dic['y_err'] = 0.0