python判断素数函数形式