python - 第二个元素大于阈值的元组拆分列表

Question

我有一个元组列表，需要将其拆分为“元组列表列表”，其中元组中的每个第二个元素都大于 1。从这些元组列表中，我尝试计算新列表中第一个元组的第一个元素与新列表中最后一个元组的第一个元素之间的差异。抱歉，如果这看起来令人困惑。

以下为例:

zp = [(1.31, 0), (1.32, 0),
      (1.33, 0), (1.34, 0),
      (1.35, 1), (1.36, 2),
      (1.37, 3), (1.37, 4),
      (1.39, 5), (1.38, 6),
      (1.40, 7), (1.41, 8),
      (1.42, 9), (1.43, 10),
      (1.44, 0), (1.45, 0),
      (1.46, 0), (1.47, 0),
      (1.48, 1), (1.49, 2),
      (1.50, 3), (1.51, 4),
      (1.52, 0), (1.53, 0),
      (1.54, 1), (1.55, 2),
      (1.56, 3), (1.57, 4),
      (1.48, 0), (1.59, 0),
      (1.60, 0), (1.61, 0),
      (1.62, 1), (1.63, 2),
      (1.64, 3), (1.65, 4),
      (1.66, 0), (1.67, 0),
      (1.68, 1), (1.69, 2),
      (1.70, 3), (1.71, 4)]

从这里我枚举元组第二个元素 > 1 的所有位置，并获取 zp 的索引序列:

enums = [j for j, k in enumerate(zp) if not k[1] in (0,1)]

enums 
[5, 6, 7, 8, 9, 10, 11, 12, 13, 19, 20, 21, 25, 26, 27, 33, 34, 35, 39, 40, 41]

从这里我开始怀疑我走错了方向，因为我已经绕了好几个圈了。例如，使用 for 循环获取 zp 的切片:

for k in range(1, len(enums)):
    if enums[k] - enums[k-1] > 1:
        print "index k-1", k-1, "end if seq: emum[k-1]", enums[k-1], " --", "index k", k, "start if next seq: emum[k-1]", enums[k]

然后从那里开始，使用更改后的 enum 索引对 zp 进行切片...所以必须有更好的方法，对吗？

欢迎任何建议。

*编辑

预期结果:

[(1.36, 2),(1.37, 3), (1.37, 4),(1.39, 5), (1.38, 6),(1.40, 7), (1.41, 8),(1.42, 9), (1.43, 10)],
[(1.49, 2),(1.50, 3), (1.51, 4)], 
[(1.55, 2),(1.56, 3), (1.57, 4)],
[(1.63, 2),(1.64, 3), (1.65, 4)],
[(1.69, 2),(1.70, 3), (1.71, 4)]]

Answer 1

您可以使用itertools.groupby根据每个元组的第二项 > 1 (lambda x: x[1] > 1) 的条件对结果进行分组。这将为条件具有相同结果(True 或 False)的每组连续项目返回一个组。然后，我们可以使用列表理解来忽略因条件为 False 而分组的组，并将这些组转换为条件为 True 的列表。

import itertools
A = [list(group) for val, group in itertools.groupby(zp, lambda x: x[1] > 1) if val]

# [[(1.36, 2), (1.37, 3), (1.37, 4), (1.39, 5), (1.38, 6), (1.4, 7), (1.41, 8), (1.42, 9), (1.43, 10)], 
#  [(1.49, 2), (1.5, 3), (1.51, 4)],
#  [(1.55, 2), (1.56, 3), (1.57, 4)], 
#  [(1.63, 2), (1.64, 3), (1.65, 4)], 
#  [(1.69, 2), (1.7, 3), (1.71, 4)]]