魯棒加權隨機字符串生成器

Question

我已經實現了以下類，根​​據輸入頻率'p'生成'p'或'q'。但是，如果頻率小於用於存儲選項的列表大小，則此實現會中斷。有沒有一種方法可以實現這一點，以便爲p的任何值工作？

from random import random 

class AlleleGenerator(object): 
    """ 
    allele generator - will break if p < 0.001 
    """ 
    def __init__(self, p): 
     """construct class and creates list to select from""" 
     self.values = list() 
     for i in xrange(int(1000*p)): 
      self.values.append('p') 
     while len(self.values) <= 1000: 
      self.values.append('q') 
    def next(self): 
     """Returns p or q based on allele frequency""" 
     rnd = int(random() * 1000) 
     return self.values[rnd] 
    def __call__(self): 
     return self.next() 

Answer 1

請勿使用self.values。在next，只是生成0和1之間的隨機數，並返回'p'如果隨機數小於p：

from random import random 

class AlleleGenerator(object): 
    def __init__(self, p): 
     """construct class and creates list to select from""" 
     self.p = p 
    def next(self): 
     """Returns p or q based on allele frequency"""    
     return 'p' if random() < self.p else 'q' 
    def __call__(self): 
     return self.next() 

此外，小心not to use classes當一個功能就足夠了。 例如，你可以考慮使用一個generator function：

from random import random 

def allele_generator(p): 
    while True: 
     yield 'p' if random() < p else 'q' 

agen = allele_generator(0.001) 
for i in range(3): 
    print(next(agen))