Pythonic 切片列表的方式 w.r.t.元组中的第一个元素

Question

我有一个形式为元组的排序列表

x = 
[(0,1), (0,2), (0,3), ... 
 (1,1), (1,3), (1,4), ...
 ...
 (n,0), (n,4), ...
]

我想对列表进行切片，以便 (x,y) 的所有数字（其中 x 是新列表中的某个值）并保持顺序。现在，这显然会起作用：

y = [(a,b) for (a,b) in x if a == n]

但它真的很慢。使用二进制搜索找到满足此条件的第一个和最后一个索引会更快。 index 为您提供值的第一个索引，而反向列表的 index 将提供最后一个索引。如果不做[a for (a,b) in x] 并以pythonic 方式复制整个列表，如何应用它？

Answer 1

正如@Liongold 在 cmets 中所建议的，您可以使用 bisect。假设您想要所有元组 t 和 t[0] == 1：

from bisect import bisect_left

x = [(0, 1), (0, 2), (1, 1), (1, 2), (2, 1), (2, 2)]

start = bisect_left(x, (1, None))  # index of the very first (1, i) 
end = bisect_left(x, (2, None))  # index after the very last (1, i)

y = x[start:end]

# y: [(1, 1), (1, 2)]

您可以在bisect docs找到详细信息。