python实现一个简单的猜数游戏

要求：预设一个0-9之间的整数，让用户猜一猜并输入所猜的数，如果大于预设的数，显示“太大”；小于预设的数，显示“太小”；如此循环，直至猜中该数，显示“恭喜！你猜中了！”

代码：

def guessNum(guess):
    '''
    guess: 要猜的数
    num： 输入的数
    '''
    num = int(input())   
    if guess == num:
        print('恭喜！你猜中了！')
    else:
        if num > guess:
            print('太大')
            return guessNum(guess)
        else:
            print('太小')
            return guessNum(guess)

if __name__ == "__main__":
    guess = int(input('预设一个0-9之间的整数:')) # 预设一个0-9之间的整数
    guessNum(guess)