python - 如何迭代整数对列表，计算新的 'union pairs'

Question

我有一个表示年份范围的整数对列表，需要计算连续(1 年内)对的联合范围。

输入示例:

ts_list = [[1777, 1777], [1778, 1783], [1786, 1791], [1792, 1795]]

所需输出

[[1777, 1781], [1786, 1795]]

我已经尝试过 for 和 while 循环，并且可以在第一次不相交之前获得并集，但我对如何正确迭代感到困惑 - 例如这会产生一个新列表

[[1777, 1783], [1778, 1783], [1786, 1795]]

然后返回类型错误:'int'对象不可下标”。第一对和第三对是正确的，但第二对是无关的

ts_list = [[1777, 1777], [1778, 1781], [1786, 1791], [1792, 1795]]
newlist=[]
last = ts_list[len(ts_list)-1][1]
for x in range(len(ts_list)):
   ts=ts_list[x]
   start = ts[0]
   end = ts[1]
   ts_next = ts_list[x+1] if x<len(ts_list)-1 else last
   if ts_next[0]-end > 1:
       # next is disjoint, break out
       newlist.append([start,end])
   else:
       # next is contiguous (within 1 year)
       newlist.append([start,ts_next[1]])

Answer 1

你可以这样做:

ts_list = [[1777, 1777], [1778, 1781], [1786, 1791], [1792, 1795]]

# We start with the first range
out = [ts_list[0]]
for start, end in ts_list[1:]:
    if start <= out[-1][1] + 1:
        # if the new range starts at most one year
        # after the end of the previous one, we extend it:
        out[-1][1] = end
    else:
        # otherwise, we append this new range to the output
        out.append([start, end])

print(out)

# [[1777, 1781], [1786, 1795]]