红篮球游戏

"""
红篮球游戏
Red basketball game
RedBasketballGame.py
"""
#   引入
import random
import pandas as pd


# 创建类
class ReadBlueBoll(object):

    # 外部传参
    def __init__(self, red_nums, blue_nums, times, red_start_range, red_end_range, blue_start_range,  blue_end_range):
        self.red_nums = red_nums
        self.blue_nums = blue_nums
        self.times = times
        self.red_start_range = red_start_range
        self.red_end_range = red_end_range
        self.blue_start_range = blue_start_range
        self.blue_end_range = blue_end_range

    @staticmethod
    def themselves(self):
        for _ in range(self.times):
            print(f'\033[30m第{_ +1}组红篮球数据展示\033[0m')
            # 生成随机红球号码
            red_balls = random.sample(range(self.red_start_range, self.red_end_range), self.red_nums)

            # 生成随机蓝球号码
            blue_ball = random.sample(range(self.blue_start_range, self.blue_end_range), self.blue_nums)

            # 打印结果

            print(f"随机红球: \033[31m{red_balls}\033[0m")
            print(f"随机蓝球: \033[34m{blue_ball}\033[0m")

        # 假设你的代码输出结果是一个字典
        result = {
            '红球': [self.red_balls],
            '蓝球': [self.blue_balls]
        }

        # 将字典转换为 DataFrame
        rb = pd.DataFrame(result)

        # 将 DataFrame 内容写入 Excel 文件中
        rb.to_excel('output.xlsx', index=False)


# 实现类
rb = ReadBlueBoll(15, 3, 5, 1, 100, 1, 55)
rb.themselves(rb)