Python - 前 K 个唯一元素

在Python编程中，有时需要从一个列表、数组或集合中获取前k个唯一元素。这通常是在数据分析、机器学习和人工智能领域等需要数据的操作中使用的。

以下是如何使用Python获取前K个唯一元素的示例代码：

def get_top_k_unique_elements(arr, k):
    unique_elems = []
    duplicates = set()
    for elem in arr:
        if elem not in duplicates:
            if len(unique_elems) == k:
                break
            unique_elems.append(elem)
            duplicates.add(elem)
    return unique_elems

代码解释：

• 定义一个空列表和一个空集合用于存储唯一元素和重复元素。
• 迭代列表中的每个元素。
• 如果元素不在重复元素中，则将其添加到唯一元素列表中，并添加到重复元素集合中。
• 如果唯一元素列表的长度达到了K，则停止迭代。
• 返回唯一元素列表。

该函数返回的是一个长度为K且包含唯一元素的列表。如果输入列表中不足K个唯一元素，则返回长度小于K的列表。

下面是使用示例：

arr = [1, 2, 2, 3, 4, 5, 5, 5, 6, 6]
k = 3
print(get_top_k_unique_elements(arr, k))

输出：

[1, 2, 3]

在这个示例中，列表中有3个唯一元素（1、2、3），而我们需要返回3个唯一元素，因此返回的是[1, 2, 3]。

此代码可以轻松实现在Python中获取前K个唯一元素的功能。