returnall=True)
indsnot = q2.filterbydistancefromline(comps,
compend1,
compend2,
critdist,
betweenpoints=betweenpoints,
invlogic=True)
print len(inds), ' points'
q2.scatter(comps[inds], c=comps[inds, 3])
q2.scatter(comps[indsnot], c='grey', marker='.', s=5)
q2.line(compend1, compend2)
q2.label(ha='center', va='center', fontsize=16)
q2.set_projection(azim=-17, elev=-6)
pylab.figure()
lp = lineparameter[inds]
argsinds = numpy.argsort(lp)
pylab.plot(lp[argsinds], comps[inds, 3][argsinds], 'b-')
pylab.plot(lp[argsinds], comps[inds, 3][argsinds], 'bo')
ax = pylab.gca()
lineparticks = numpy.linspace(0, 1, 4)
tl = []
for i in lineparticks:
c = compend1 + (compend2 - compend1) * i
tl += [q.singlelabeltext(c)]
ax.xaxis.set_ticks(lineparticks)
ax.xaxis.set_ticklabels(tl)
pylab.show()
lp=lined['lineparameter'][code0inds]
# lp[lp<0.]=0.
# lp[lp>1.]=1.
fom=fomall[code0inds]
argsinds=numpy.argsort(lp)
pylab.plot(lp[argsinds], fom[argsinds], 'b-')
pylab.plot(lp[argsinds], fom[argsinds], 'bo')
if not linedsmoothfcn is None:
fomsmooth=linedsmoothfcn(lp[argsinds], fom[argsinds])
pylab.plot(lp[argsinds], fomsmooth, 'k-')
axlinepar=pylab.gca()
lineparticks=numpy.linspace(0, 1, 4)
tl=[]
for i in lineparticks:
c=lined['compend1']+(lined['compend2']-lined['compend1'])*i
tl+=[stpquat.singlelabeltext(c)]
pylab.xlim(0, 1)
#pylab.ylim(vmin, vmax)
axlinepar.xaxis.set_ticks(lineparticks)
axlinepar.xaxis.set_ticklabels(tl)
pylab.ylabel(fomlabel)
if SYSTEM==1:
axbin.set_ylim(.23, .7)
if SYSTEM==6:
axbin.set_ylim(.38, .5)
if not os.path.exists(savefolder):
os.mkdir(savefolder)
os.chdir(savefolder)
q.label(ha='center', va='center', fontsize=16)
q.set_projection(azim=-17, elev=-6)
inds, distfromlin, lineparameter=q2.filterbydistancefromline(comps, compend1, compend2, critdist, betweenpoints=betweenpoints, invlogic=False, returnall=True)
indsnot=q2.filterbydistancefromline(comps, compend1, compend2, critdist, betweenpoints=betweenpoints, invlogic=True)
print len(inds), ' points'
q2.scatter(comps[inds],c=comps[inds,3])
q2.scatter(comps[indsnot],c='grey', marker='.', s=5)
q2.line(compend1, compend2)
q2.label(ha='center', va='center', fontsize=16)
q2.set_projection(azim=-17, elev=-6)
pylab.figure()
lp=lineparameter[inds]
argsinds=numpy.argsort(lp)
pylab.plot(lp[argsinds], comps[inds,3][argsinds], 'b-')
pylab.plot(lp[argsinds], comps[inds,3][argsinds], 'bo')
ax=pylab.gca()
lineparticks=numpy.linspace(0, 1, 4)
tl=[]
for i in lineparticks:
c=compend1+(compend2-compend1)*i
tl+=[q.singlelabeltext(c)]
ax.xaxis.set_ticks(lineparticks)
ax.xaxis.set_ticklabels(tl)
pylab.show()
