编写函数sushu（）来判断一个大于等于2的正整数是否是素数，如果是返回true，不是则返回flase。输入一个正整数呢，输出大于等于n的第3小素数。python

sushu代码：

def sushu(n):
    if n < 2:
        return False
    for i in range(2, int(n**0.5)+1):
        if n % i == 0:
            return False
    return True

输出大于等于n的第3小素数的代码：

def third_sushu(n):
    count = 0
    i = n
    while True:
        if sushu(i):
            count += 1
            if count == 3:
                return i
        i += 1

请问还有什么可以帮助您的吗？