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 x,y= np.loadtxt('saxs.data',usecols=(0,1),unpack=True)
  13 x1,y1= np.loadtxt('MODEL1_saxs.data',usecols=(0,2),unpack=True)
  14 x2,y2= np.loadtxt('MODEL2_saxs.data',usecols=(0,2),unpack=True)
  15 x3,y3= np.loadtxt('MODEL3_saxs.data',usecols=(0,2),unpack=True)
  16 x4,y4= np.loadtxt('MODEL4_saxs.data',usecols=(0,2),unpack=True)
  17 x5,y5= np.loadtxt('MODEL5_saxs.data',usecols=(0,2),unpack=True)
  18
  19
  20 plt.clf()
  21 plt.xlabel('r')
  22 plt.ylabel('P(r)')
  23 plt.xlim(x[0], x[-1])
  24 plt.plot(x,y,'-',color='black')
  25 plt.plot(x1,y1,'-',color='red')
  26 plt.plot(x2,y2,'-',color='lime')
  27 plt.plot(x3,y3,'-',color='blue')
  28 plt.plot(x4,y4,'-',color='magenta')
  29 plt.plot(x5,y5,'-',color='cyan')
  30 if os.path.isfile('plik_saxs.data'):
  31   x6,y6= np.loadtxt('plik_saxs.data',usecols=(0,2),unpack=True)
  32   plt.plot(x6,y6,'--',color='black')
  33   plt.legend(['reference','MODEL1','MODEL2','MODEL3','MODEL4','MODEL5','input pdb'])
  34 else:
  35   plt.legend(['reference','MODEL1','MODEL2','MODEL3','MODEL4','MODEL5'])
  36 plt.savefig('saxs.png')