Python算法-素数判断优化