使用 python 查找列表中唯一的最大值

Question

我有一个点列表，如下所示，

points=[ [x0,y0,v0],  [x1,y1,v1],  [x2,y2,v2].......... [xn,yn,vn]]

其中一些点具有重复的 x,y 值。我想要做的是提取唯一的最大值 x,y 点

例如，如果我有点 [1,2,5] [1,1,3] [1,2,7] [1, 7,3]

我想获取列表[1,1,3] [1,2,7] [1,7,3]

我如何在 python 中执行此操作？

谢谢

Answer 1

例如：

import itertools

def getxy(point): return point[:2]

sortedpoints = sorted(points, key=getxy)

results = []

for xy, g in itertools.groupby(sortedpoints, key=getxy):
  results.append(max(g, key=operator.itemgetter(2)))

即：按 xy 对点进行排序和分组，对于具有固定 xy 的每个组，选取具有最大 z 的点。如果您熟悉 itertools (你应该如此，它确实是一个非常强大且有用的模块！）。

或者，您可以构建一个字典，其中 (x,y) 元组作为键，z 列表作为值，然后对该字典进行最后一次传递以选择最大 z 对于每个 (x, y)，但我认为排序和分组方法更可取（除非你有数百万个点，以至于排序的大 O 性能令人担忧我猜是出于可扩展性的目的）。

Answer 2

您可以使用 dict 来实现此目的，使用 属性表示“如果给定的键出现多次，则与其关联的最后一个值将保留在新字典中。”此代码对点进行排序以确保最高值出现在后面，创建一个字典，其键是前两个值的元组，其值是第三个坐标，然后将其转换回列表

points = [[1,2,5], [1,1,3], [1,2,7], [1,7,3]]
sp = sorted(points)
d = dict( ( (a,b), c) for (a,b,c) in sp)
results = [list(k) + [v] for (k,v) in d.iteritems()]

可能有一种方法可以进一步改进它，但它满足您的所有要求。

Answer 3

如果我理解你的问题..也许使用字典将 (x,y) 映射到 max z

像这样（未测试）

dict = {}
for x,y,z in list
    if dict.has_key((x,y)):
        dict[(x,y)] = max(dict[(x,y)], z)
    else:
        dict[(x,y)] = z

虽然顺序会丢失