python itertools模块实现排列组合

2018年1月23日 发表评论 阅读评论

在使用python写一个简单字典密码生成器的时候,了解到了itertools模块和笛卡尔积(一个数学生成式)。发现其对于字典生成太有帮助了。

一、你会怎么写?

如果不考虑这个模块自身,我们自己来实现,会怎么实现一个字典生成功能呢?看下我写的一个示例代码:

#!/usr/bin/env python
# code from www.361way.com
import string
basestring = string.ascii_lowercase + string.digits
for s1 in string.ascii_lowercase:
    for s2 in basestring:
        print (s1 + s2)

但是上面只是针对,两位的字符串组合,如果是三位、四位、五位……呢?----- 写一个基于两位的函数,然后在这个基础循环,还是通过random.choice模块随机选择?貌似可以随机进行选择,但是肯定会产生重复值,而且也不一定能random出所有值 。接下来看下itertools的使用。

二、笛卡尔积:itertools.product(*iterables[, repeat])

直接对自身进行笛卡尔积:

import itertools
for i in itertools.product('ABCD', repeat = 2):
    print (''.join(i),end=' ')

输出结果:
AA AB AC AD BA BB BC BD CA CB CC CD DA DB DC DD 

 print (”.join(i))这个语句可以让结果直接排列到一起 

 end=’ ‘可以让默认的输出后换行变为一个空格。

两个元组进行笛卡尔积:

import itertools
a = (1, 2, 3)
b = ('A', 'B', 'C')
c = itertools.product(a,b)
for i in c:
    print(i,end=' ')

输出结果: (1, ‘A’) (1, ‘B’) (1, ‘C’) (2, ‘A’) (2, ‘B’) (2, ‘C’) (3, ‘A’) (3, ‘B’) (3, ‘C’)

三、排列:itertools.permutations(iterable[, r])

import itertools
for i in itertools.permutations('ABCD', 2):
    print (''.join(i),end=' ')

输出结果: AB AC AD BA BC BD CA CB CD DA DB DC

四、组合:itertools.combinations(iterable, r)

import itertools
for i in itertools.combinations('ABCD', 3):
    print (''.join(i))

输出结果:

ABC
ABD
ACD
BCD

五、组合(包含自身重复):itertools.combinations_with_replacement(iterable, r)

import itertools
for i in itertools.combinations_with_replacement('ABCD', 3):
    print (''.join(i),end=' ')

输出结果: AAA AAB AAC AAD ABB ABC ABD ACC ACD ADD BBB BBC BBD BCC BCD BDD CCC CCD CDD DDD




本站的发展离不开您的资助,金额随意,欢迎来赏!

You can donate through PayPal.
My paypal id: itybku@139.com
Paypal page: https://www.paypal.me/361way

  1. 本文目前尚无任何评论.
  1. 本文目前尚无任何 trackbacks 和 pingbacks.