python求素数的函数