我有一个按升序排序的列表列表,类似于这个:
input = [[1,1],[1,2],[1,3],[1,4],[2,1],[2,2],[2,3],[3,1],[6,1],[6,2]]
我想过滤此列表,以便新列表只包含位置0中匹配整数的前两个(或唯一)元素,如下所示:
output = [[1,1],[1,2],[2,1],[2,2],[3,1],[6,1],[6,2]]
如果其余元素(不符合标准的元素)将保留在输入列表中,而匹配元素将单独存储,那将是理想的.
我该怎么做呢？
先感谢您!
编辑:索引1上的元素实际上可以是任何整数,例如 [[1,6],[1,7],[1,8],[2,1],[2,2]]