python模型i.i.d贝努利试验

我正在尝试建立独立的伯努利试验模型。这是试图模拟一个患流感的人在一个房间里给10个人注射（这是一个虚构的场景）。我使用此函数尝试对其进行建模：

p = 0.01 #given
n = 10 #given

def perform_bernoulli_trials(n, p):
    """
    Perform n Bernoulli trials with success probability p
    and return number of successes.
    """
    # Initialize number of successes: n_success
    n_success = 0

    # Perform trials
    for i in range(n):
        # Choose random number between zero and one: random_number
        random_number = np.random.random()

        # If less than p, it's a success so add one to n_success
        if random_number < p:
            n_success += 1

    return n_success

result = perform_bernoulli_trials(n, p) #pretend result = 1

谢谢！请让我知道思考过程是否正确，或者是否有更好的方法编写此函数

perform_bernoulli_trials(9, 0.01) + perform_bernoulli_trials(9, 0.01)