Question

我想要一个字符串列表：

[2000q1, 2000q2, 2000q3, 2000q4,
 2001q1, 2001q2, 2001q3, 2001q4,
 2002q1, 2002q2, 2002q3, 2002q4 ...]

等等。

我想通过str.format在Python中创建上面的结果。

以下是我的尝试：

import numpy as np
x = list(range(2000, 2003))
x = np.repeat(x, 4)
y = [1,2,3,4] * 3

"{}q{}".format(x,y)
# One string containing all numbers (FAILED)

"{x[i for i in x]}q{y[j for j in y]}".format(**{"x": x, "y": y})
# IndexError (FAILED)

最后我通过以下方式解决了这个问题：

result = list()
for i in range(0, len(y)):
    result.append("{}q{}".format(x[i],y[i]))
result

还有更优雅的解决方案，不需要显式循环吗？我在R中寻找这样的东西：

sprintf("%dq%d", x, y)

Answer 1

您可以使用嵌套列表理解：

result = ['{}q{}'.format(y, q+1) for y in range(2000, 2003) for q in range(4)]

Answer 2

您可以使用map来获得功能性的，但更为丑陋的解决方案：

import itertools
final_data = list(itertools.chain(*map(lambda x:map(lambda y:"{}q{}".format(x, y), range(1, 5)), range(2000, 2003))))

输出：

['2000q1', '2000q2', '2000q3', '2000q4', '2001q1', '2001q2', '2001q3', '2001q4', '2002q1', '2002q2', '2002q3', '2002q4']

使用数字序列格式化字符串以生成由Python中的数字排序的字符串列表

2 个答案: