4.8:编写程序,输出所有由1、2、3、4这4个数字组成的素数,并且在每个素数中每个数字只能使用一次。
思路:1、编写判断是否为素数的函数isPrime() 2、生成全排列的所有可能,对其逐一进行判断,符合条件即输出
from itertools import permutations
digits = (1,2,3,4)
def isPrime(n):
if n==1:
return False
if n == 2:
return True
if n%2 == 0:
return False
for i in range(3,int(n**0.5)+1,2):
if n%i == 0:
return False
return True
for i in range(1,len(digits)+1):
for num in permutations(digits,i):
num = int(''.join(map(str,num)))
if isPrime(num):
print(num)
itertools.permutations() 第一个参数是接收一个课迭代的对象,第二个参数指定每次排列时从迭代对象(第一个参数)中选着几个字符进行排列。也可以不传入第二个参数,那么默认就是可迭代对象的长度,最终返回生成元组的迭代器
例如:
>>> import itertools
>>> x = itertools.permutations([1,2,3,4],4)
>>> print(x)
<itertools.permutations object at 0x00000142D78ED620>
>>> print(*x)
(1, 2, 3, 4) (1, 2, 4, 3) (1, 3, 2, 4) (1, 3, 4, 2) (1, 4, 2, 3) (1, 4, 3, 2) (2, 1, 3, 4) (2, 1, 4, 3) (2, 3, 1, 4) (2, 3, 4, 1) (2, 4, 1, 3) (2, 4, 3, 1) (3, 1, 2, 4) (3, 1, 4, 2) (3, 2, 1, 4) (3, 2, 4, 1) (3, 4, 1, 2) (3, 4, 2, 1) (4, 1, 2, 3) (4, 1, 3, 2) (4, 2, 1, 3) (4, 2, 3, 1) (4, 3, 1, 2) (4, 3, 2, 1)