根据条件组合列表中的列表

Question

我正在尝试创建一个 for 循环来根据条件组合列表中的列表。

我的数据集有 300 条宠物记录。有 3 种动物（例如狗、猫、鸟），每个动物出生年份的间隔（例如（1990、2000]、（2000、2004] 等）和友好仪表的间隔为每只动物评分（例如 (0,2]、(2, 4]、(4, 6] 等））

可以在此处找到数据集外观的示例图像： https://i.stack.imgur.com/YwbGN.png

我用来组合友好仪表评级的当前代码如下：

friendly_meter_combos = []
combo = []

for val in df['friendly_meter'].unique():
    if len(combo) < 2:
        combo.append(val)
    else:
        friendly_meter_combos.append(combo)
        combo = []
        combo.append(val)

这为我提供了

的示例结果

friendly_meter_combos = 
[[(0, 2], (2, 4]],
 [(4, 6], (6, 8]],
 [(8, 10], (2, 4]], 
 [(4, 6], (6, 8]],
 ...]. 

正如您在我打印的倒数第二行中看到的那样，数字从头开始 ([(8, 10], (2, 4]])。这是因为我现在处于不同的年份间隔为数据集中相同的动物类型而生。我想弄清楚如何重新开始代码以获得下面的输出

friendly_meter_combos = 
[[(0, 2], (2, 4]],
 [(4, 6], (6, 8]],
 [(8, 10]], 
 [(2, 4], (4, 6]],
 ...]. 

任何帮助将不胜感激！谢谢！

Answer 1

据我了解，您希望获得连续“友好仪表”的组合

# let's suppose unique_values is a list containing ordered friendly meters
unique_values = [ [0, 2], [2, 4], [4, 6], [6, 8], [8, 10] ]
# n is the size of the combo 
n = 2
# use statement below
friendly_meter_combos = list(zip(*(unique_values[i:] for i in range(n))))

结果是： [([0, 2], [2, 4]), ([2, 4], [4, 6]), ([4, 6], [6, 8]), ([6, 8], [8, 10])]