时间戳

首页 > 解决方案 > 字母大小写排列

python - 字母大小写排列

问题描述

我正在尝试获取字符串的所有字母大小写排列并将它们添加到一个列表中，如下所示：

allperm('an') should produce:
['an', 'aN', 'An', 'AN']

但我的输出是多个列表：

['a', 'n']
['a', 'N']

这是我当前的代码

def allperm(password):
   n = len(password)

   #2^n permuations
   max = 1 << n

   #lowercase password (base)
   password = password.lower()

   #get all permutations
   for i in range(max):
       perm  = [k for k in password]
       for j in range(n):
           if (((i >> j) & 1) == 1):
               perm[j] = password[j].upper()

       #add to list
     temp = ""
      for i in perm:
           temp += i
      return(temp)

标签： pythonpermutation

解决方案

您可以zip将密码的小写字符与大写字符一起使用，并将它们传递itertools.product给以生成所需的排列：

from itertools import product
def allperm(password):
    return list(map(''.join, product(*zip(password.lower(), password.upper()))))

这样allperm('an')返回：

['an', 'aN', 'An', 'AN']

推荐阅读