import numpy as np import random from Distribution import Distribution class ProbabilitySet: def __init__(self,dset,p): self.dset = dset self.probiablity = p def dataset(self): dl = Distribution([1,0],[self.probiablity,]) v = dl.generator(len(self.dset)) m = [] for i in range(len(self.dset)): if v[i] == 1: m.append(self.dset) return m if __name__ == '__main__': from timecost import TimeCost dset = range(10000000) pp = [0.1,0.22,0.334,0.88,0.93] tc = TimeCost() for p in pp: tc.begin('dataset() %f' % p) ps = ProbablitySet(dset,p) d = ps.dataset() tc.end('dataset() %f' % p) tc.show()