如何用自定义比较函数实现more_itertools.all_unique的自定义唯一性判断？

架构老林

2026-8-9

用自定义等价关系检查列表元素唯一性的实现方案

核心思路

由于你的自定义eq是可交换的等价关系（满足自反、对称、传递性），本质上可以将每个元素映射到一个「等价类标识」——当且仅当两个元素的标识相等时，eq(a,b)为真。基于这个特性，我们可以用两种高效方式实现需求：

方法1：标准库集合（推荐，无需额外依赖）

构造一个与eq对应的key函数，将每个元素转换为等价类标识，再通过集合去重后对比长度：

# 示例自定义等价关系：两字符串等价当且仅当长度相同
def eq(a, b):
    return len(a) == len(b)

# 对应key函数：提取等价类标识
key_func = lambda x: len(x)

# 检查唯一性
items = ["apple", "banana", "cherry"]
is_unique = len({key_func(x) for x in items}) == len(items)
# 输出：False（"apple"和"cherry"长度均为5，属于同一等价类）

方法2：利用more_itertools库

如果已经依赖more_itertools，可以用unique_everseen函数结合key参数，对比去重前后的列表长度：

from more_itertools import unique_everseen

# 沿用上面的eq和key_func
items = ["apple", "banana", "cherry"]
unique_list = list(unique_everseen(items, key=key_func))
is_unique = len(unique_list) == len(items)

方法3：直接使用eq函数（适合小列表）

如果不想构造key函数，直接用itertools.combinations遍历所有元素对判断：

from itertools import combinations

def all_unique(items, eq):
    # 检查所有两两元素对是否都不满足等价关系
    return not any(eq(a, b) for a, b in combinations(items, 2))

# 调用示例
items = ["apple", "banana", "cherry"]
is_unique = all_unique(items, eq)

注意：这种方法时间复杂度为O(n²)，仅适合元素数量较少的场景。