如何循环遍历列表并使用“zip()”将一项添加到元组列表中的每个元组？

Question

我的数据文件包含八千行，其中每一行都是一个元组。前两行是：

(2000, 1, 1, 1, 336)

(2000, 1, 1, 2, 335)

我想重复循环工作日并将工作日附加到每个数据元组。行列表应如下所示：

[(2000, 1, 1, 1, 336, 'Sat'), (2000, 1, 1, 2, 335, 'Sun'), ...]

我尝试了不同的方法，并设法获得了除所需组合之外的几乎所有可能的组合。我无法完全理解我所误解的事情。

代码

weekdays = ['Sat','Sun','Mon','Tue','Wed','Thu','Fri']    

a = read_file('VIK_sealevel_2000.txt')

zipp = list(zip(a, weekdays))

print(zipp[0:2])   

输出

[((2000, 1, 1, 1, 335), 'Sat'), ((2000, 1, 1, 2, 336), 'Sun')]

Answer 1

第一个问题是如何高效地循环度过工作日。一个干净的解决方案是使用itertools.cycle().以下将生成无限重复的工作日序列。import itertoolsweekdays = ['Sat','Sun','Mon','Tue','Wed','Thu','Fri']itertools.cycle(weekdays)zip()当最短输入可迭代用完时，该函数将停止。现在我们有无限的工作日序列，最短的输入将是八千行数据。>>> import itertools>>> weekdays = ['Sat','Sun','Mon','Tue','Wed','Thu','Fri']>>> data = [(2000, 1, 1, 1, 336), (2000, 1, 1, 2, 335), (2000, 1, 1, 1, 334),             (2000, 1, 1, 2, 333), (2000, 1, 1, 1, 332), (2000, 1, 1, 2, 331),             (2000, 1, 1, 1, 330), (2000, 1, 1, 2, 329), (2000, 1, 1, 1, 328),             (2000, 1, 1, 2, 327)]>>> list(zip(data, itertools.cycle(weekdays)))[((2000, 1, 1, 1, 336), 'Sat'), ((2000, 1, 1, 2, 335), 'Sun'),  ((2000, 1, 1, 1, 334), 'Mon'), ((2000, 1, 1, 2, 333), 'Tue'),  ((2000, 1, 1, 1, 332), 'Wed'), ((2000, 1, 1, 2, 331), 'Thu'),  ((2000, 1, 1, 1, 330), 'Fri'), ((2000, 1, 1, 2, 329), 'Sat'),  ((2000, 1, 1, 1, 328), 'Sun'), ((2000, 1, 1, 2, 327), 'Mon')]最后，为了获取内部元组中的工作日，我们可以在列表理解中创建新的元组。>>> [t + (day,) for t, day in zip(data, itertools.cycle(weekdays))][(2000, 1, 1, 1, 336, 'Sat'), (2000, 1, 1, 2, 335, 'Sun'),  (2000, 1, 1, 1, 334, 'Mon'), (2000, 1, 1, 2, 333, 'Tue'),  (2000, 1, 1, 1, 332, 'Wed'), (2000, 1, 1, 2, 331, 'Thu'),  (2000, 1, 1, 1, 330, 'Fri'), (2000, 1, 1, 2, 329, 'Sat'),  (2000, 1, 1, 1, 328, 'Sun'), (2000, 1, 1, 2, 327, 'Mon')]这里，t是与每行数据关联的元组，day是工作日，并且语法t + (day,)创建通过将数据元组与工作日组合形成的新元组。

Answer 2

要将工作日添加到每个元组中，请尝试使用：a = [(2000, 1, 1, 1, 335), (2000, 1, 1, 2, 336)] # was read_file('VIK_sealevel_2000.txt')weekdays = ['Sat','Sun','Mon','Tue','Wed','Thu','Fri']zipp = [row[0] + (row[1],) for row in zip(a, weekdays)]