将一个列表的每个字符串元素连接到另一个列表的所有元素

Question

year = ['2019', '2020', '2021', '2022']
week = [str(i).zfill(2) for i in range(1,53)]

我正在尝试将年份列表的每个字符串元素加入/连接到周列表的所有元素，如下所示

['201901'
'201902'
'201903'
.
.
.
.
.
'202250'
'202251'
'202252']

• Output 应该是一个列表

我可以使用以下代码实现这一点

week = [str(i).zfill(2) for i in range(1,53)]
year = [str(i) for i in range(2019,2023)]*len(range(2019,2023))*52
year.sort()
print({i:year.count(i) for i in year})

li = []
for yr in set(year):
    li.append([i + j for i, j in zip([k for k in year if yr in k], week)])
yr_week_id = sorted(list(np.concatenate(li)))
print(yr_week_id)

这个问题有没有更简单的解决方案？

Answer 1

你真的不需要周列表。 您可以像 go 一样计算周数。

如果年份范围可以用range()表示，您甚至可能不需要年份列表

year = ['2019', '2020', '2021', '2022']
output = []
for yy in year:
    output.extend([f'{yy}{ww:02d}' for ww in range(1, 53)])
print(output)

Output：

['201901',
'201902',
...
'202251',
'202252']