help with plt.fill_between

[maria]

Hi,

I want to write this script in a shorter way maybe using do loops.
Also in the way the script is written the instruction plt.fill_between is giving me
the following error: ValueError: Argument dimensions are incompatible

Any help? Also how do I attach a file to this message?

Maria

%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.ticker import AutoMinorLocator
from pylab import *

data_chi = np.loadtxt('energies.dat')
data_il7 = np.loadtxt('IL7.dat')
data_exp = np.loadtxt('exp.dat')

xx_chi = data_chi[:,0]
xx_il7 = data_il7[:,0]
xx_exp = data_exp[:,0]

xx_chi = xx_chi +3.
xx_il7 = xx_il7 +3. +0.1
xx_exp = xx_exp +3. +0.2


x_chi = [xx_chi,xx_chi+0.4]
x_il7 = [xx_il7,xx_il7+0.4]
x_exp = [xx_exp,xx_exp+0.4]


yy_chi_gs = data_chi[:,1]
ey_chi_gs = data_chi[:,2]
yy_il7_gs = data_il7[:,1]
ey_il7_gs = data_il7[:,2]
yy_exp_gs = data_exp[:,1]
ey_exp_gs = data_exp[:,2]

yy_chi_1s = data_chi[:,3]
ey_chi_1s = data_chi[:,4]
yy_il7_1s = data_il7[:,3]
ey_il7_1s = data_il7[:,4]
yy_exp_1s = data_exp[:,3]
ey_exp_1s = data_exp[:,4]

yy_chi_2s = data_chi[:,5]
ey_chi_2s = data_chi[:,6]
yy_il7_2s = data_il7[:,5]
ey_il7_2s = data_il7[:,6]
yy_exp_2s = data_exp[:,5]
ey_exp_2s = data_exp[:,6]

yy_chi_3s = data_chi[:,7]
ey_chi_3s = data_chi[:,8]
yy_il7_3s = data_il7[:,7]
ey_il7_3s = data_il7[:,8]
yy_exp_3s = data_exp[:,7]
ey_exp_3s = data_exp[:,8]

yy_chi_4s = data_chi[:,9]
ey_chi_4s = data_chi[:,10]
yy_il7_4s = data_il7[:,9]
ey_il7_4s = data_il7[:,10]
yy_exp_4s = data_exp[:,9]
ey_exp_4s = data_exp[:,10]

yy_chi_5s = data_chi[:,11]
ey_chi_5s = data_chi[:,12]
yy_il7_5s = data_il7[:,11]
ey_il7_5s = data_il7[:,12]
yy_exp_5s = data_exp[:,11]
ey_exp_5s = data_exp[:,12]

lw=0.7

y_chi_gs  = [yy_chi_gs,yy_chi_gs]
yl_chi_gs = [yy_chi_gs-ey_chi_gs,yy_chi_gs-ey_chi_gs]
yu_chi_gs = [yy_chi_gs+ey_chi_gs,yy_chi_gs+ey_chi_gs]
plt.plot(x_chi,yu_chi_gs,color='blue',linewidth=lw)
plt.plot(x_chi,y_chi_gs,color='blue',linewidth=lw)
plt.plot(x_chi,yl_chi_gs,color='blue',linewidth=lw)

y_chi_1s  = [yy_chi_1s,yy_chi_1s]
yl_chi_1s = [yy_chi_1s-ey_chi_1s,yy_chi_1s-ey_chi_1s]
yu_chi_1s = [yy_chi_1s+ey_chi_1s,yy_chi_1s+ey_chi_1s]
plt.plot(x_chi,yu_chi_1s,color='blue',linewidth=lw)
plt.plot(x_chi,y_chi_1s,color='blue',linewidth=lw)
plt.plot(x_chi,yl_chi_1s,color='blue',linewidth=lw)

y_chi_2s  = [yy_chi_2s,yy_chi_2s]
yl_chi_2s = [yy_chi_2s-ey_chi_2s,yy_chi_2s-ey_chi_2s]
yu_chi_2s = [yy_chi_2s+ey_chi_2s,yy_chi_2s+ey_chi_2s]
plt.plot(x_chi,yu_chi_2s,color='blue',linewidth=lw)
plt.plot(x_chi,y_chi_2s,color='blue',linewidth=lw)
plt.plot(x_chi,yl_chi_2s,color='blue',linewidth=lw)

y_chi_3s  = [yy_chi_3s,yy_chi_3s]
yl_chi_3s = [yy_chi_3s-ey_chi_3s,yy_chi_3s-ey_chi_3s]
yu_chi_3s = [yy_chi_3s+ey_chi_3s,yy_chi_3s+ey_chi_3s]
plt.plot(x_chi,yu_chi_3s,color='blue',linewidth=lw)
plt.plot(x_chi,y_chi_3s,color='blue',linewidth=lw)
plt.plot(x_chi,yl_chi_3s,color='blue',linewidth=lw)

y_chi_4s  = [yy_chi_4s,yy_chi_4s]
yl_chi_4s = [yy_chi_4s-ey_chi_4s,yy_chi_4s-ey_chi_4s]
yu_chi_4s = [yy_chi_4s+ey_chi_4s,yy_chi_4s+ey_chi_4s]
plt.plot(x_chi,yu_chi_4s,color='blue',linewidth=lw)
plt.plot(x_chi,y_chi_4s,color='blue',linewidth=lw)
plt.plot(x_chi,yl_chi_4s,color='blue',linewidth=lw)

y_chi_5s  = [yy_chi_5s,yy_chi_5s]
yl_chi_5s = [yy_chi_5s-ey_chi_5s,yy_chi_5s-ey_chi_5s]
yu_chi_5s = [yy_chi_5s+ey_chi_5s,yy_chi_5s+ey_chi_5s]
plt.plot(x_chi,yu_chi_5s,color='blue',linewidth=lw)
plt.plot(x_chi,y_chi_5s,color='blue',linewidth=lw)
plt.plot(x_chi,yl_chi_5s,color='blue',linewidth=lw)

y_il7_gs  = [yy_il7_gs,yy_il7_gs]
yl_il7_gs = [yy_il7_gs-ey_il7_gs,yy_il7_gs-ey_il7_gs]
yu_il7_gs = [yy_il7_gs+ey_il7_gs,yy_il7_gs+ey_il7_gs]
plt.plot(x_il7,yu_il7_gs,color='red',linewidth=lw)
plt.plot(x_il7,y_il7_gs,color='red',linewidth=lw)
plt.plot(x_il7,yl_il7_gs,color='red',linewidth=lw)

y_il7_1s  = [yy_il7_1s,yy_il7_1s]
yl_il7_1s = [yy_il7_1s-ey_il7_1s,yy_il7_1s-ey_il7_1s]
yu_il7_1s = [yy_il7_1s+ey_il7_1s,yy_il7_1s+ey_il7_1s]
plt.plot(x_il7,yu_il7_1s,color='red',linewidth=lw)
plt.plot(x_il7,y_il7_1s,color='red',linewidth=lw)
plt.plot(x_il7,yl_il7_1s,color='red',linewidth=lw)

y_il7_2s  = [yy_il7_2s,yy_il7_2s]
yl_il7_2s = [yy_il7_2s-ey_il7_2s,yy_il7_2s-ey_il7_2s]
yu_il7_2s = [yy_il7_2s+ey_il7_2s,yy_il7_2s+ey_il7_2s]
plt.plot(x_il7,yu_il7_2s,color='red',linewidth=lw)
plt.plot(x_il7,y_il7_2s,color='red',linewidth=lw)
plt.plot(x_il7,yl_il7_2s,color='red',linewidth=lw)

y_il7_3s  = [yy_il7_3s,yy_il7_3s]
yl_il7_3s = [yy_il7_3s-ey_il7_3s,yy_il7_3s-ey_il7_3s]
yu_il7_3s = [yy_il7_3s+ey_il7_3s,yy_il7_3s+ey_il7_3s]
plt.plot(x_il7,yu_il7_3s,color='red',linewidth=lw)
plt.plot(x_il7,y_il7_3s,color='red',linewidth=lw)
plt.plot(x_il7,yl_il7_3s,color='red',linewidth=lw)

y_il7_4s  = [yy_il7_4s,yy_il7_4s]
yl_il7_4s = [yy_il7_4s-ey_il7_4s,yy_il7_4s-ey_il7_4s]
yu_il7_4s = [yy_il7_4s+ey_il7_4s,yy_il7_4s+ey_il7_4s]
plt.plot(x_il7,yu_il7_4s,color='red',linewidth=lw)
plt.plot(x_il7,y_il7_4s,color='red',linewidth=lw)
plt.plot(x_il7,yl_il7_4s,color='red',linewidth=lw)

y_il7_5s  = [yy_il7_5s,yy_il7_5s]
yl_il7_5s = [yy_il7_5s-ey_il7_5s,yy_il7_5s-ey_il7_5s]
yu_il7_5s = [yy_il7_5s+ey_il7_5s,yy_il7_5s+ey_il7_5s]
plt.plot(x_il7,yu_il7_5s,color='red',linewidth=lw)
plt.plot(x_il7,y_il7_5s,color='red',linewidth=lw)
plt.plot(x_il7,yl_il7_5s,color='red',linewidth=lw)

y_exp_gs  = [yy_exp_gs,yy_exp_gs]
yl_exp_gs = [yy_exp_gs-ey_exp_gs,yy_exp_gs-ey_exp_gs]
yu_exp_gs = [yy_exp_gs+ey_exp_gs,yy_exp_gs+ey_exp_gs]
plt.plot(x_exp,yu_exp_gs,color='green',linewidth=lw)
plt.plot(x_exp,y_exp_gs,color='green',linewidth=lw)
plt.plot(x_exp,yl_exp_gs,color='green',linewidth=lw)

y_exp_1s  = [yy_exp_1s,yy_exp_1s]
yl_exp_1s = [yy_exp_1s-ey_exp_1s,yy_exp_1s-ey_exp_1s]
yu_exp_1s = [yy_exp_1s+ey_exp_1s,yy_exp_1s+ey_exp_1s]
plt.plot(x_exp,yu_exp_1s,color='green',linewidth=lw)
plt.plot(x_exp,y_exp_1s,color='green',linewidth=lw)
plt.plot(x_exp,yl_exp_1s,color='green',linewidth=lw)

y_exp_2s  = [yy_exp_2s,yy_exp_2s]
yl_exp_2s = [yy_exp_2s-ey_exp_2s,yy_exp_2s-ey_exp_2s]
yu_exp_2s = [yy_exp_2s+ey_exp_2s,yy_exp_2s+ey_exp_2s]
plt.plot(x_exp,yu_exp_2s,color='green',linewidth=lw)
plt.plot(x_exp,y_exp_2s,color='green',linewidth=lw)
plt.plot(x_exp,yl_exp_2s,color='green',linewidth=lw)

y_exp_3s  = [yy_exp_3s,yy_exp_3s]
yl_exp_3s = [yy_exp_3s-ey_exp_3s,yy_exp_3s-ey_exp_3s]
yu_exp_3s = [yy_exp_3s+ey_exp_3s,yy_exp_3s+ey_exp_3s]
plt.plot(x_exp,yu_exp_3s,color='green',linewidth=lw)
plt.plot(x_exp,y_exp_3s,color='green',linewidth=lw)
plt.plot(x_exp,yl_exp_3s,color='green',linewidth=lw)

y_exp_4s  = [yy_exp_4s,yy_exp_4s]
yl_exp_4s = [yy_exp_4s-ey_exp_4s,yy_exp_4s-ey_exp_4s]
yu_exp_4s = [yy_exp_4s+ey_exp_4s,yy_exp_4s+ey_exp_4s]
plt.plot(x_exp,yu_exp_4s,color='green',linewidth=lw)
plt.plot(x_exp,y_exp_4s,color='green',linewidth=lw)
plt.plot(x_exp,yl_exp_4s,color='green',linewidth=lw)

y_exp_5s  = [yy_exp_5s,yy_exp_5s]
yl_exp_5s = [yy_exp_5s-ey_exp_5s,yy_exp_5s-ey_exp_5s]
yu_exp_5s = [yy_exp_5s+ey_exp_5s,yy_exp_5s+ey_exp_5s]
plt.plot(x_exp,yu_exp_5s,color='green',linewidth=lw)
plt.plot(x_exp,y_exp_5s,color='green',linewidth=lw)
plt.plot(x_exp,yl_exp_5s,color='green',linewidth=lw)


#plt.fill_between(x_exp, yl_exp_5s, yu_exp_5s, facecolor='yellow', alpha=0.5)
plt.fill_between(x_exp, yl_exp_5s, yu_exp_5s, alpha=0.5, facecolor='b',edgecolor = 'none')
#plt.show()
plt.axis([0, 40, -100, -0.2])
#plt.xticks([])
ml = MultipleLocator(4)
plt.axes().yaxis.set_minor_locator(ml) 
ftz=12
plt.text(4.0, -18,'$^3$H', fontsize=ftz)
plt.text(7.0, -18,'$^3$He', fontsize=ftz)
plt.text(2.5, -38,'$^4$He', fontsize=ftz)
plt.text(6.0, -38,'$^6$He', fontsize=ftz)
plt.text(9.8, -42,'$^6$Li', fontsize=ftz)
plt.text(12.8, -48,'$^7$Li', fontsize=ftz)
plt.text(15.5,-40,'$^8$He', fontsize=ftz)
plt.text(19.0,-49,'$^8$Li', fontsize=ftz)
plt.text(22.0,-65,'$^8$Be', fontsize=ftz)
plt.text(25.0,-55,'$^9$Li', fontsize=ftz)
plt.text(27.0,-68,'$^9$Be', fontsize=ftz)
plt.text(30.0,-75,'$^{10}$Be', fontsize=ftz)
plt.text(34.0,-75,'$^{10}$B', fontsize=ftz)
savefig('spectrum.pdf')