在python中生成数字和交替字符串的序列

Question

目标

我想在python生成一个序列作为list ，例如：

['s1a', 's1b', 's2a', 's2b', ..., 's10a', 's10b']

特性：

• 项目包含单个前缀
• 数字按数字排序
• 后缀按数字交替显示

方法

为此，我使用了xrange和综合列表方法应用了以下代码：

# prefix
p = 's'
# suffix
s = ['a', 'b']
# numbers
n = [ i + 1 for i in list(xrange(10))]
# result
[ p + str(i) + j for i, j in zip(sorted(n * len(s)), s * len(n)) ]

题

是否有更简单的语法来获取结果，例如使用itertools ？ 类似于这个问题？

Answer 1

双重理解列表可以实现以下目的：

['s'+str(x)+y for x in range(1,11) for y in 'ab']

Answer 2

itertools.product可能是您的朋友：

all_combos = ["".join(map(str, x)) for x in itertools.product(p, n, s)]

返回：

['s1a', 's1b', 's2a', 's2b', 's3a', 's3b', 's4a', 's4b', 's5a', 's5b', 's6a', 's6b', 's7a', 's7b', 's8a', 's8b', 's9a', 's9b', 's10a', 's10b']

编辑：作为单线：

all_combos = ["".join(map(str,x)) for x in itertools.product(['s'], range(1, 11), ['a', 'b'])]

编辑2：正如詹姆斯回答中所指出的，我们可以将product调用中列出的字符串元素更改为仅字符串，并且itertools仍然可以对其进行迭代，并从每个字符串中选择字符：

all_combos = ["".join(map(str,x)) for x in itertools.product('s', range(1, 11), 'ab')]

Answer 3

怎么样：

def func(prefix,suffixes,size):
    k = len(suffixes)
    return [prefix+str(n/k+1)+suffixes[n%k] for n in range(size*k)]

# usage example:
print func('s',['a','b'],10)

这样，您可以根据需要替换任意数量的后缀。

当然，每个后缀都可以根据需要设置。

Answer 4

您可以使用双列表理解，在其中迭代数字和后缀。 您不需要加载任何

下面是一个lambda函数，它带有3个参数，一个前缀，多次迭代和一个后缀列表

foo = lambda prefix,n,suffix: list(prefix+str(i)+s for s in suffix for i in range(n))

你可以这样使用

foo('p',10,'abc')

或者这样，如果您的后缀有多个字母

foo('p',10,('a','bc','de'))

Answer 5

为了获得最大的通用性，我会以此为发​​电机。 这样，您既可以创建列表，也可以仅根据需要生成序列项。

这是在Python 2或Python 3上运行的代码。

def psrange(prefix, suffix, high):
    return ('%s%d%s' % (prefix, i, s) for i in range(1, 1 + high) for s in suffix)

res = list(psrange('s', ('a', 'b'), 10))
print(res)

for s in psrange('x', 'abc', 3):
    print(s)

输出

['s1a', 's1b', 's2a', 's2b', 's3a', 's3b', 's4a', 's4b', 's5a', 's5b', 's6a', 's6b', 's7a', 's7b', 's8a', 's8b', 's9a', 's9b', 's10a', 's10b']
x1a
x1b
x1c
x2a
x2b
x2c
x3a
x3b
x3c