Python寻找第N个默尼森数

寻找第n个默尼森数。 代码格式如下： def prime(num): … def monisen(no): … … return xxx print(monisen(int(input()))) #此处不需要自己输入，只要写这样一条语句即可，主要完成monisen()函数（4分）
经典程序设计问题：找第n个默尼森数。P是素数且M也是素数，并且满足等式M=2**P-1，则称M为默尼森数。例如，P=5，M=2**P-1=31，5和31都是素数，因此31是默尼森数。

输入格式:按提示用input()函数输入

输出格式：int类型

输入样例：4

输出样例：127

时间限制：500ms内存限制：32000kb
python3代码：

from math import sqrt


def prime(num):
    if 0 not in [num % d for d in range(2, int(sqrt(num))+1)]:
        return 1
    else:
        return 0


def monisen(no):
    flag = 2
    x = 1
    a = []
    a.append(0)
    while x - 1 != no:
        if prime(flag) == 1:
            temp = 2**flag-1
            if prime(temp) == 1:
                a.append(temp)
                x += 1
        flag += 1
    return a[no]

print(monisen(int(input())))