files/plot_saxs.py

   1 #! /usr/bin/env python
   2
   3 import matplotlib
   4 #matplotlib.use('GTK')
   5 matplotlib.use('Agg')
   6 import matplotlib.pyplot as plt
   7 import matplotlib.cm as cm
   8 import numpy as np
   9 import sys
  10 import os.path
  11
  12 matplotlib.rc('font', size=15)
  13 matplotlib.rc('legend', fontsize=15)
  14
  15 x,y= np.loadtxt('saxs.data',usecols=(0,1),unpack=True)
  16 x1,y1= np.loadtxt('MODEL1_saxs.data',usecols=(0,2),unpack=True)
  17 x2,y2= np.loadtxt('MODEL2_saxs.data',usecols=(0,2),unpack=True)
  18 x3,y3= np.loadtxt('MODEL3_saxs.data',usecols=(0,2),unpack=True)
  19 x4,y4= np.loadtxt('MODEL4_saxs.data',usecols=(0,2),unpack=True)
  20 x5,y5= np.loadtxt('MODEL5_saxs.data',usecols=(0,2),unpack=True)
  21
  22
  23 plt.clf()
  24 plt.xlabel(r'r [$\AA$]')
  25 plt.ylabel('P(r)')
  26 plt.xlim(x[0], x[-1])
  27 plt.plot(x,y,'-',color='black')
  28 plt.plot(x1,y1,'-',color='red')
  29 plt.plot(x2,y2,'-',color='lime')
  30 plt.plot(x3,y3,'-',color='blue')
  31 plt.plot(x4,y4,'-',color='magenta')
  32 plt.plot(x5,y5,'-',color='cyan')
  33 if os.path.isfile('plik_saxs.data'):
  34   x6,y6= np.loadtxt('plik_saxs.data',usecols=(0,2),unpack=True)
  35   plt.plot(x6,y6,'--',color='black')
  36   plt.legend(['reference','MODEL1','MODEL2','MODEL3','MODEL4','MODEL5','input pdb'])
  37 else:
  38   plt.legend(['reference','MODEL1','MODEL2','MODEL3','MODEL4','MODEL5'])
  39 plt.savefig('saxs.png')