import numpy as np
import matplotlib.pyplot as plt

plt.rcParams['axes.linewidth'] = 2
plt.rcParams['xtick.major.size']=8
plt.rcParams['ytick.major.size']=8
plt.rcParams['xtick.major.width']=2
plt.rcParams['ytick.major.width']=2
plt.rcParams['xtick.direction']='in'
plt.rcParams['ytick.direction']='in'

Data = np.loadtxt('TD_Absorption.dat')

plt.plot(Data[:,0],Data[:,1])
plt.xlabel('Time [fs]',fontsize=16)
plt.ylabel('Response function [arb.u.]',fontsize=16)
plt.show()
plt.savefig('TD_absorption')

Data = np.loadtxt('Absorption.dat')

plt.plot(Data[:,0],Data[:,1])
plt.xlabel('$\omega$ [cm$^{-1}$]',fontsize=16)
plt.ylabel('Absorption [arb.u.]',fontsize=16)
plt.show()
plt.savefig('absorption.png')
