用Python制作像Mastermind这样的游戏

import random

def generate_number():
    number = random.sample(range(10), 4)
    return number

def compare_numbers(answer, guess):
    result = {'A': 0, 'B': 0}
    for i in range(len(answer)):
        if guess[i] == answer[i]:
            result['A'] += 1
        elif guess[i] in answer:
            result['B'] += 1
    return result

answer = generate_number()
guess_count = 0

while True:
    guess = input("请输入一个四位数字：")
    guess = [int(i) for i in guess]
    guess_count += 1
    
    result = compare_numbers(answer, guess)
    print("结果：{}A{}B".format(result['A'], result['B']))
    
    if result['A'] == 4:
        print("恭喜你猜对了！答案是{}".format(answer))
        break
    
    if guess_count == 10:
        print("很遗憾，你没有猜对。答案是{}".format(answer))
        break