zachelpme

import numpy as np
import matplotlib.pyplot as plt

#For all of the spectral functions, w21=-1. This allows the displacement between the peak and satellite to be the same for each graph
w = [] #Establishes x-axis in Frequency Space
a_w_0_25 = [] #Spectral Function when v=0.25
a_w_0_5 = []
a_w_0_75 = []
a_w_1 = []
a_w_1_5 = []
a_w_2 = []
a_w_3 = []
a_w_5 = []
z_0_25 = [] # Main Peak Integrated Weight at v = 0.25
z_0_5 = []
z_0_75 = []
z_1 = []
z_1_5 = []
z_2 = []
z_3 = []
z_5 = []

data_0_25 = np.genfromtxt('0_25.dat', dtype = None, delimiter = '')
data_0_5 = np.genfromtxt('ugh.dat', dtype = None, delimiter = '')
data_0_75 = np.genfromtxt('0_75.dat', dtype = None, delimiter = '')
data_1 = np.genfromtxt('1.dat', dtype = None, delimiter = '')
data_1_5 = np.genfromtxt('1_5.dat', dtype = None, delimiter = '')
data_2 = np.genfromtxt('2.dat', dtype = None, delimiter = '')
data_3 = np.genfromtxt('3.dat', dtype = None, delimiter = '')
data_5 = np.genfromtxt('5.dat', dtype = None, delimiter = '')

for row in data_0_25:
    w.append(row[0])
    a_w_0_25.append(row[1])
    z_0_25.append(row[2])
for row in data_0_5:
    a_w_0_5.append(row[1])
    z_0_5.append(row[2])
for row in data_0_75:
    a_w_0_75.append(row[1])
    z_0_75.append(row[2])
for row in data_1:
    a_w_1.append(row[1])
    z_1.append(row[2])
for row in data_1_5:
    a_w_1_5.append(row[1])
    z_1_5.append(row[2])
for row in data_2:
    a_w_2.append(row[1])
    z_2.append(row[2])
for row in data_3:
    a_w_3.append(row[1])
    z_3.append(row[2])
for row in data_5:
    a_w_5.append(row[1])
    z_5.append(row[2])


plt.plot(w,a_w_0_25,'b')
#plt.plot(w,a_w_0_5)
plt.plot(w,a_w_0_75,'g')
plt.plot(w,a_w_1,'m')
plt.plot(w,a_w_1_5,'c')
plt.plot(w,a_w_2,'y')
plt.plot(w,a_w_3,'bo')
plt.plot(w,a_w_5,'ro')
plt.xlabel('Frequency w')
plt.ylabel('A(w)')
plt.grid(True)
plt.show()