python - 如何枚举组合字符串作为 for..in 循环的搜索范围？

Question

我编写了一个对我有意义但对 python 没有意义的代码，因为我是 python 新手。

在这里检查我的代码：

checksum_algos = ['md5','sha1']

for filename in ["%smanifest-%s.txt" % (prefix for prefix in ['', 'tag'],  a for a in checksum_algos)]:
  f = os.path.join(self.path, filename)
  if isfile(f):
     yield f

我的意图是在如下列表中搜索文件名：

['manifest-md5.txt','tagmanifest-md5.txt','manifest-sha1.txt','tagmanifest-sha1.txt']

但我在syntax实施它时遇到了问题。

谢谢你的帮助。

Answer 1

你想多了。

for filename in ("%smanifest-%s.txt" % (prefix, a)
    for prefix in ['', 'tag'] for a in checksum_algos):

Answer 2

使用新样式字符串格式和itertools：

from itertools import product
["{0}manifest-{1}.txt".format(i,e) for i,e in  product(*(tags,checksum_algos))]

出去：

['manifest-md5.txt',
 'manifest-sha1.txt',
 'tagmanifest-md5.txt',
 'tagmanifest-sha1.txt']

Answer 3

或者你需要itertools.product()：

>>> import itertools

>>> [i for i in itertools.product(('', 'tag'), ('sha', 'md5'))]
[('', 'sha'), ('', 'md5'), ('tag', 'sha'), ('tag', 'md5')]