Difference between revisions of "Matplotlib disp spin.out"

From relax wiki
Jump to navigation Jump to search
Line 11: Line 11:
 
== Code ==
 
== Code ==
 
<source lang="python">
 
<source lang="python">
 +
import os
 
import sys
 
import sys
 
import numpy as np
 
import numpy as np
Line 21: Line 22:
  
 
infile = sys.argv[1]
 
infile = sys.argv[1]
 +
filename , fileext = os.path.splitext(infile)
 
print("Input file is: %s"%infile)
 
print("Input file is: %s"%infile)
  
Line 60: Line 62:
 
plt.grid(True)
 
plt.grid(True)
 
plt.title("%s as function of %s"%(fileindex[3], fileindex[2]))
 
plt.title("%s as function of %s"%(fileindex[3], fileindex[2]))
 +
plt.savefig("matplotlib_%s.png"%filename)
 +
  
 
plt.show()
 
plt.show()
</source>
 
 
To run
 
<source lang="bash">
 
python -i testmatplotlib.py disp_13_N.out
 
python -i testmatplotlib.py disp_theta_13_N.out
 
 
</source>
 
</source>
  
 
== See also ==
 
== See also ==
 
[[Category:Matplotlib]]
 
[[Category:Matplotlib]]

Revision as of 15:25, 13 March 2014

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]

# 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