按概率值筛选对象

import random

class ObjectWithProbability:
    def __init__(self, name, probability):
        self.name = name
        self.probability = probability

def select_object(objects):
    rand_value = random.random()  # 生成一个[0, 1)之间的随机数
    cumulative_probability = 0
    
    for obj in objects:
        cumulative_probability += obj.probability
        if rand_value < cumulative_probability:
            return obj

# 示例对象列表
objects = [
    ObjectWithProbability("A", 0.1),
    ObjectWithProbability("B", 0.3),
    ObjectWithProbability("C", 0.2),
    ObjectWithProbability("D", 0.4)
]

# 选择对象
selected_object = select_object(objects)
print(f"Selected object: {selected_object.name}")