027. 编写一个函数，检查一个列表中的所有元素是否相同

在 Python 中，可以通过多种方式检查一个列表中的所有元素是否相同。以下是一个示例函数，展示如何实现这一功能。

示例代码

def are_all_elements_same(lst):
    """
    检查一个列表中的所有元素是否相同。

    参数:
        lst (list): 要检查的列表。

    返回:
        bool: 如果所有元素相同，返回 True；否则返回 False。
    """
    if not lst:  # 如果列表为空，返回 True
        return True

    # 获取列表中的第一个元素
    first_element = lst[0]

    # 使用 all() 函数检查所有元素是否与第一个元素相同
    return all(element == first_element for element in lst)

# 测试代码
list1 = [1, 1, 1, 1]
list2 = [1, 2, 1, 1]
list3 = []

print(are_all_elements_same(list1))  # 输出：True
print(are_all_elements_same(list2))  # 输出：False
print(are_all_elements_same(list3))  # 输出：True

代码解释

检查空列表：

• 如果列表为空，函数返回 True，因为空列表可以认为是所有元素相同（没有元素）。

获取第一个元素：

• 使用 lst[0] 获取列表中的第一个元素。

使用 all() 函数：

• all() 函数用于检查可迭代对象中的所有元素是否都满足某个条件。

• 使用生成器表达式 element == first_element for element in lst 遍历列表中的每个元素，并检查它们是否与第一个元素相同。

• 如果所有元素都与第一个元素相同，all() 函数返回 True；否则返回 False。

扩展：使用集合

另一种方法是将列表转换为集合，然后检查集合的长度是否为 1。如果集合的长度为 1，说明列表中的所有元素都相同。

def are_all_elements_same(lst):
    """
    检查一个列表中的所有元素是否相同。

    参数:
        lst (list): 要检查的列表。

    返回:
        bool: 如果所有元素相同，返回 True；否则返回 False。
    """
    return len(set(lst)) <= 1

# 测试代码
list1 = [1, 1, 1, 1]
list2 = [1, 2, 1, 1]
list3 = []

print(are_all_elements_same(list1))  # 输出：True
print(are_all_elements_same(list2))  # 输出：False
print(are_all_elements_same(list3))  # 输出：True

代码解释

将列表转换为集合：

• 使用 set(lst) 将列表转换为集合，集合中的元素是唯一的。

检查集合的长度：

• 如果集合的长度为 1 或 0（空列表），说明列表中的所有元素都相同。

注意事项

空列表：

• 空列表可以认为是所有元素相同，因此函数返回 True。

性能：

• 使用 all() 函数的方法在遍历过程中一旦发现不相同的元素就会停止，效率较高。

• 使用集合的方法需要将整个列表转换为集合，可能会占用更多内存，但代码更简洁。

视频讲解

BiliBili： 视睿网络-哔哩哔哩视频 (bilibili.com)