在一定高度上收集所有点的最有效方法是什么?

我有一个 List[List[int]] 表示点[x,y]

points = [[1,2],[1,3],[2,1],[2,3]]

收集具有特定高度的所有点的最有效方法是什么?

因此,如果我们想要所有高度为3(y==3)的点:

answer = [[1,3],[2,3]]

我想还有一种比这更有效的方法:

answer = []
for p in points:
    if p[1] == 3:
        answer.append(p)

此外,如何查询至少有一个点的不同高度集?

answer = [1,2,3]