plot_pickle.py

import pickle
import matplotlib.pyplot as plt
import numpy

fig = plt.figure()
fig.show()
ax = fig.add_subplot(111)


# get the data

linestyles=['--',':','-.','-','-:']
for charge in [12,13,14,15,16]:
  dat = pickle.load(open('spec_v1_ne1cm3_S_%i.pkl'%(charge),'rb'))
  
  for iE, E in enumerate(dat['Elist']):
    # these are the electron beam energies
  
    s = dat['spec'][iE,:]
    s = numpy.append(s[0],s)
    
    if iE == 0:
  
      ls, = ax.plot(dat['ebins'], s, label=repr(charge)+' '+repr(E), drawstyle='steps', linestyle=linestyles[iE])
    
    else:
      ax.plot(dat['ebins'], s, label=repr(charge)+' '+repr(E), drawstyle='steps', color=ls.get_color(), linestyle=linestyles[iE])
  
  ax.legend(loc=0)

plt.draw()
zzz=input()