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

remoce matplotlib import

Browse Source
This commit is contained in:
Dominik Demuth
2023-04-16 14:33:05 +02:00
parent 48861c3f80
commit bb3d5ac58b
1 changed files with 63 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

127
src/nmreval/io/fcbatchreader.py
View File

@@ -2,7 +2,6 @@ from __future__ import annotations
import pathlib
from matplotlib import pyplot as plt
# import matplotlib.pyplot as plt
from scipy.optimize import curve_fit
import numpy as np
@@ -219,16 +218,16 @@ class FCReader:
np.savetxt(fit_path.joinpath(save_name), np.c_[xplot, yplot],
header=header+'\t'.join([f'{p}+/-{err}' for p, err in zip(p0, perr)]))
if save_fig:
fig, ax = plt.subplots()
ax.set_xlabel('t / s')
ax.set_ylabel('M')
axheader = f'T1: {p0[2]:.4g}(+/-{perr[2]:.4g}) beta: {p0[3]:.4g}(+/-{perr[3]:.4g})'
ax.set_title(f'f = {k:.4g} Hz\n{axheader}')
ax.semilogx(v.x, v.y, 'o')
ax.semilogx(xplot, yplot, '-')
fig.savefig(image_path.joinpath(save_name).with_suffix('.png'))
plt.close(fig)
# if save_fig:
# fig, ax = plt.subplots()
# ax.set_xlabel('t / s')
# ax.set_ylabel('M')
# axheader = f'T1: {p0[2]:.4g}(+/-{perr[2]:.4g}) beta: {p0[3]:.4g}(+/-{perr[3]:.4g})'
# ax.set_title(f'f = {k:.4g} Hz\n{axheader}')
# ax.semilogx(v.x, v.y, 'o')
# ax.semilogx(xplot, yplot, '-')
# fig.savefig(image_path.joinpath(save_name).with_suffix('.png'))
# plt.close(fig)
freqs = np.asanyarray(freqs)
params = np.asanyarray(params)
@@ -278,16 +277,16 @@ class FCReader:
np.savetxt(save_path.joinpath(f'fitparameter_{key}.dat'), parameter,
header=f'{key}\nM0\tM0_err\tOff\tOff_err\tT1\tT1_err\tbeta\tbeta_err')
@staticmethod
def _plot_parameter(key, param, fig_mag, fig_t1, fig_beta):
ax_mag = fig_mag.get_axes()[0]
ax_t1 = fig_t1.get_axes()[0]
ax_beta = fig_beta.get_axes()[0]
pl, = ax_mag.plot(param[:, 0], param[:, 1], 'o', label=key)
ax_mag.plot(param[:, 0], param[:, 3], 's', color=pl.get_color())
ax_t1.plot(param[:, 0], param[:, 5], 'o', label=key)
ax_beta.plot(param[:, 0], param[:, 7], 'o', label=key)
# @staticmethod
# def _plot_parameter(key, param, fig_mag, fig_t1, fig_beta):
# ax_mag = fig_mag.get_axes()[0]
# ax_t1 = fig_t1.get_axes()[0]
# ax_beta = fig_beta.get_axes()[0]
#
# pl, = ax_mag.plot(param[:, 0], param[:, 1], 'o', label=key)
# ax_mag.plot(param[:, 0], param[:, 3], 's', color=pl.get_color())
# ax_t1.plot(param[:, 0], param[:, 5], 'o', label=key)
# ax_beta.plot(param[:, 0], param[:, 7], 'o', label=key)
@staticmethod
def _save_parameter_plot(path, kind, fig_mag, fig_t1, fig_beta):
@@ -316,49 +315,49 @@ class FCReader:
# plt.close(fig_beta)
# plt.close(fig_t1)
def plot_parameter(self, path, kind):
path = pathlib.Path(path)
path.mkdir(parents=True, exist_ok=True)
fig_mag, ax_mag = plt.subplots()
fig_t1, ax_t1 = plt.subplots()
fig_beta, ax_beta = plt.subplots()
if kind == 'temp':
_params = self.t_params
else:
_params = self.f_params
save_path = path.joinpath(kind)
if not save_path.exists():
save_path.mkdir(parents=True)
for key, par in _params.items():
pl, = ax_mag.plot(par[:, 0], par[:, 1], 'o', label=key)
ax_mag.plot(par[:, 0], par[:, 3], 's', color=pl.get_color())
ax_t1.plot(par[:, 0], par[:, 5], 'o', label=key)
ax_beta.plot(par[:, 0], par[:, 7], 'o', label=key)
for a in [ax_mag, ax_t1, ax_beta]:
if kind == 'freq':
a.legend(loc='upper left', bbox_to_anchor=(1, 1), ncol=2)
a.set_xlabel('T / K')
else:
a.set_xscale('log')
a.legend(loc='upper left', bbox_to_anchor=(1, 1))
a.set_xlabel('f / Hz')
ax_t1.set_yscale('log')
ax_t1.set_ylabel('T1 / s')
ax_beta.set_ylabel('beta')
ax_mag.set_ylabel('M0 (squares), Offset (circles)')
fig_beta.savefig(path.joinpath(f'beta_{kind}.png'), bbox_inches="tight")
fig_mag.savefig(path.joinpath(f'mag_{kind}.png'), bbox_inches="tight")
fig_t1.savefig(path.joinpath(f't1_{kind}.png'), bbox_inches="tight")
plt.close(fig_mag)
plt.close(fig_beta)
plt.close(fig_t1)
# def plot_parameter(self, path, kind):
# path = pathlib.Path(path)
# path.mkdir(parents=True, exist_ok=True)
#
# fig_mag, ax_mag = plt.subplots()
# fig_t1, ax_t1 = plt.subplots()
# fig_beta, ax_beta = plt.subplots()
#
# if kind == 'temp':
# _params = self.t_params
# else:
# _params = self.f_params
#
# save_path = path.joinpath(kind)
# if not save_path.exists():
# save_path.mkdir(parents=True)
#
# for key, par in _params.items():
# pl, = ax_mag.plot(par[:, 0], par[:, 1], 'o', label=key)
# ax_mag.plot(par[:, 0], par[:, 3], 's', color=pl.get_color())
# ax_t1.plot(par[:, 0], par[:, 5], 'o', label=key)
# ax_beta.plot(par[:, 0], par[:, 7], 'o', label=key)
#
# for a in [ax_mag, ax_t1, ax_beta]:
# if kind == 'freq':
# a.legend(loc='upper left', bbox_to_anchor=(1, 1), ncol=2)
# a.set_xlabel('T / K')
# else:
# a.set_xscale('log')
# a.legend(loc='upper left', bbox_to_anchor=(1, 1))
# a.set_xlabel('f / Hz')
#
# ax_t1.set_yscale('log')
# ax_t1.set_ylabel('T1 / s')
# ax_beta.set_ylabel('beta')
# ax_mag.set_ylabel('M0 (squares), Offset (circles)')
#
# fig_beta.savefig(path.joinpath(f'beta_{kind}.png'), bbox_inches="tight")
# fig_mag.savefig(path.joinpath(f'mag_{kind}.png'), bbox_inches="tight")
# fig_t1.savefig(path.joinpath(f't1_{kind}.png'), bbox_inches="tight")
# plt.close(fig_mag)
# plt.close(fig_beta)
# plt.close(fig_t1)
def get_parameter(self, parameter='all', kind='freq', path=None, write=True, plot=True):
param_list = []