python lambda实现求素数的简短代码