Example #1
0
import argparse
import matplotlib.pyplot as plt
import rheology

# usage: plotData.py datafile [noshow]

parser = argparse.ArgumentParser(description='Process command line arguments')
parser.add_argument('--datafiles', nargs='+')
parser.add_argument('--noshow', action='store_true')

args = parser.parse_args()

for data in args.datafiles:

    shear, viscosity, tmp1, tmp2 = rheology.loadExperiment(data)

    plt.plot(shear,viscosity)
    
plt.xlabel('Shear rate (1/s)')
plt.ylabel('Viscosity (Pa s)')
if not args.noshow:
    plt.show()
Example #2
0
import argparse
import numpy
import matplotlib.pyplot as plt
import rheology

# usage: plotData.py --datafile datafile [--logplot] [--noshow]

parser = argparse.ArgumentParser(description='Process command line arguments')
parser.add_argument('--datafile')
parser.add_argument('--logplot', action='store_true')
parser.add_argument('--noshow', action='store_true')

args = parser.parse_args()

shear, viscosity, plotFile, plotTitle = rheology.loadExperiment(args.datafile)

l = plotFile.split()
if args.logplot:
    l.insert(0,'logplots/')
else:
    l.insert(0,'plots/')
plotFile = ''.join(l)

if args.logplot:
    plt.loglog(shear, viscosity)
else:
    plt.plot(shear, viscosity)
plt.xlabel('Shear rate (1/s)')
plt.ylabel('Viscosity (Pa s)')
plt.title(plotTitle)
plt.savefig(plotFile)