改进Python列表理解以消除嵌套数组

您基本上是为 weeks 顺序,直到一年的总天数,因此这也适用:

weeks = range(0, sum(months), 7)

weeks_expanded 列表只是一系列重复的整数,从1开始,每个重复7次,直到 sum(months) // 7 ,所以您可以这样做:

weeks_expanded = [wn for wn in range(1, (sum(months) // 7) + 1) for __ in range(7)]

这在列表理解中使用了一个嵌套循环,只需迭代7次。您也可以使用 itertools.product() iterator 要将其展平为单个循环,请执行以下操作:

from itertools import product

weeks_expanded = [wn for wn, __ in product(range(1, (sum(months) // 7) + 1), (None,) * 7)]

这使用7的元组 None 引用添加重复,循环将忽略这些值。

两个版本都不需要 周数 存在。