print("The dimension is incorrect. Exiting.") sys.exit() if trval not in results.keys(): print(trval, " points are not present in results. Exiting.") sys.exit() for i in range(len(outrange)): iout=outrange[i] pcf=results['pcmi'][0][iout] mindex=results['pcmi'][1][iout] #axes_labels=['Model ('+outnames[iout]+')','Polynomial Surrogate'] print('======== Output # ',str(iout+1),' (QoI ',outnames[iout],') =========') print("Plotting model-vs-surrogate at ",trval," points") output=results[trval][2][:,iout] input=results[trval][0].reshape(-1,ndim) fig=io.plot_xy(input[:,d1],output,pnames[d1], outnames[iout],label=trval+' points',savefig='xy_'+pnames[d1]+'_'+outnames[iout]+'.eps') ngr=100 #prange=np.loadtxt('prange.dat',ndmin=2) #input_grid_phys=np.linspace(prange[d,0],prange[d,1],ngr) input_grid=np.linspace(-1,1,ngr).reshape(-1,1) pc_grid=uf.func(input_grid,'PCmi',[mindex,pcf,pctype]) fig.gca().plot(input_grid,pc_grid,'r-',ms=11,label='Polynomial fit') #xdel=prange[d,1]-prange[d,0] #gca().set_xlim(prange[d,0]-0.1*xdel,prange[d,1]+0.1*xdel) legend() fig.savefig('fit_'+pnames[d1]+'_'+outnames[iout]+'_'+trval+'.eps') print('Created file fit_'+pnames[d1]+'_'+outnames[iout]+'_'+trval+'.eps')
if os.path.exists('outnames.txt'): with open('outnames.txt') as f: outnames = f.read().splitlines() outname = outnames[o] else: outname = 'Output # ' + str(o) print("Plotting %s vs %s" % (pname, outname)) inputfile = args.inputfile #sys.argv[4] outputfile = args.outputfile #sys.argv[5] input = np.loadtxt(inputfile, ndmin=2) output = np.loadtxt(outputfile, ndmin=2) io.plot_xy(input[:, d], output[:, o], pname, outname, 'xy_' + pname + '_' + outname + '.eps') # Plot output vs input1,input2 elif (plot_type == 'xxy'): d1 = int(sys.argv[2]) d2 = int(sys.argv[3]) o = int(sys.argv[4]) if os.path.exists('pnames.txt'): with open('pnames.txt') as f: pnames = f.read().splitlines() else: pnames = ['Param # ' + str(d) for d in [d1, d2]] if os.path.exists('outnames.txt'):