二维数组的random.choice，数字越大，概率越高？

import random

def weighted_choice(choices):
    # 将二维数组展开为一维数组
    flattened = [item for sublist in choices for item in sublist]
    
    # 计算权重总和
    total_weight = sum(flattened)
    
    # 生成一个随机数
    random_num = random.uniform(0, total_weight)
    
    # 根据随机数选择元素
    cumulative_weight = 0
    for item in flattened:
        cumulative_weight += item
        if random_num <= cumulative_weight:
            return item

# 示例二维数组
choices = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

# 进行随机选择
result = weighted_choice(choices)
print(result)