Python超级素数

今天同学问了我一道python超级素数的题目，

很简单很常规，但是我做的并不好，我打算梳理一下，纠正一下编码习惯。

1.质数模板更改

以前用的是当初大一学VB时的求质数模板，自以为很牛逼，实际上早已落伍且漏洞百出。
以前是这样：

def prime(x):
    for i in range(2,x):
        if(x%i==0):
            break
    return (i+1)==x

这种方法以循环变量的值来判断是否是质数，存在的问题是对于012这样的数要单独讨论，不然 i 这个变量不会被赋值，很麻烦，请忘却！
优化后是这样：

def prime(x):
    for i in range(2,x):
        if(x%i==0):
            return False
    return True

简单明了，没有用到循环变量，不会有相关的bug，是目前最简单最好理解最好记忆的模板。
倘若要稍微提升下性能，而且也不至于太难记忆，模板可以改成：

def prime(x):
    for i in range(2,int（x**0.5)+1):
        if(x%i==0):
            return False
    return True

只是改变了 i 的下限为 根号 x。

2.每n个换行输出

不要再自以为是地用 %n方法了，这样会出现的bug是第一轮换行输出会少一个。
优化为：
count从0开始，遍历到 n 则敲回车同时置count 为 0。

3.if语句的优化

不必要的情况下不要用嵌套！还是平行的好，直接编译为机器语言，速度还快。

def prime(x):
    for i in range(2,x):
        if x % i == 0:
            return False
    return True

最后的代码如下
count=0  
for i in range(2000,10000):
    if(prime(i) and prime(int(str(i)[:-1])) and prime(int(str(i)[:-2])) and prime(int(str(i)[:-3]))):
        print(i,end=" ")
        count+=1
    if(count==5):
        print()
        count=0
print()