python定义函数判断质数