如何在给定概率矩阵的情况下从二项分布中抽样

import random

def binomial_sampling(prob_matrix, num_trials):
    results = []
    for _ in range(num_trials):
        success = False
        rand_num = random.random()
        for i in range(len(prob_matrix)):
            if rand_num <= prob_matrix[i]:
                results.append(1)
                success = True
                break
            rand_num -= prob_matrix[i]
        if not success:
            results.append(0)
    return results

# 示例概率矩阵
prob_matrix = [0.2, 0.5, 0.8]

# 示例试验次数
num_trials = 10

# 进行抽样
sampling_results = binomial_sampling(prob_matrix, num_trials)

# 打印结果
print(sampling_results)