prime number impl by python

Signed-off-by: RyanZhu <troyzx1024@gmail.com>