Weighter.py 文件源码

def addDistributions(self,Tuple):
        import numpy
        selidxs=[]

        ytuple=Tuple[self.nameY]
        xtuple=Tuple[self.nameX]

        useonlyoneclass=len(self.classes)==1 and len(self.classes[0])==0

        if not useonlyoneclass:
            labeltuple=Tuple[self.classes]
            for c in self.classes:
                selidxs.append(labeltuple[c]>0)
        else:
            selidxs=[numpy.zeros(len(xtuple),dtype='int')<1]


        for i in range(len(self.classes)):
            tmphist,xe,ye=numpy.histogram2d(xtuple[selidxs[i]],ytuple[selidxs[i]],[self.axisX,self.axisY],normed=True)
            self.xedges=xe
            self.yedges=ye
            if len(self.distributions)==len(self.classes):
                self.distributions[i]=self.distributions[i]+tmphist
            else:
                self.distributions.append(tmphist)