Matplotlib disp spin.out

From relax wiki
Revision as of 15:42, 13 March 2014 by Tlinnet (talk | contribs) (→‎Code)
Jump to navigation Jump to search

About

Input file

Could look like this

# Experiment_name    Field_strength_(MHz) Tilt_angle_(rad)     R2eff_(measured)     R2eff_(back_calc)    R2eff_errors        
'R1rho'                     799.777399100            -1.530741    9.699126098118711    9.585976725687559    0.027663458874525
'R1rho'                     799.777399100            -1.544277    9.678896177797514    9.526265376687768    0.064301737965685
...

Code

import os
import sys
import numpy as np
import matplotlib.pyplot as plt

# Exit if no *.out file
if len(sys.argv) != 2:
    print("You have to provide a input .out file")
    sys.exit()

infile = sys.argv[1]
filename , fileext = os.path.splitext(infile)
print("Input file is: %s"%infile)

# Get index of file
with open(infile, 'r') as f:
    for line in f:
        if line.startswith('# Experiment_name'):
            fileindex = line.split()[1:]

# Remove not allowed letters
fileindex = [w.replace('(', '') for w in fileindex]
fileindex = [w.replace(')', '') for w in fileindex]
fileindex = [w.replace('.', '') for w in fileindex]
fileindex = [w.replace('-', '') for w in fileindex]

# Prepare to define what file exist of
fdtype="S20"
for item in fileindex[1:]:
    fdtype += ",f8"

print("Fileindex is: %s"%fileindex)
print("Value types are: %s"%fdtype)

data = np.genfromtxt(fname=infile, dtype=fdtype, names=fileindex)

xval = data[fileindex[2]]
R2eff_meas = data[fileindex[3]]
R2eff_meas_err = data[fileindex[5]]
R2eff_calc = data[fileindex[4]]

calc_not_avail = np.isnan(R2eff_calc[0])

plt.figure()
plt.errorbar(xval, R2eff_meas, yerr=R2eff_meas_err, label=fileindex[3], fmt='o')
plt.xlabel(fileindex[2])
plt.ylabel(fileindex[3])
if not calc_not_avail:
    plt.plot(xval, R2eff_calc, 'o', label=fileindex[4])
plt.legend(loc='best')
plt.grid(True)
plt.title("%s as function of %s"%(fileindex[3], fileindex[2]))
plt.savefig("matplotlib_%s.png"%filename)


plt.show()

See also