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

Merged
dominik merged 1 commits from issue195-save_format into master 2023-12-30 17:35:17 +00:00

View File

@ -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