如何高效实现判断可迭代对象元素类型单一的Python函数？

问题描述 #

我想要实现一个函数，根据输入的可迭代对象是否仅由同一类型的元素组成返回布尔值。例如，[1, 'a']返回False，[1,2,3,4]返回True。

我当前的实现代码如下：

from typing import Iterable 
def has_one_datatype(it: Iterable) -> bool:
    distinct_elements_types = {type(el) for el in it}
    try:
        assert len(distinct_elements_types) == 1
    Except AssertionError:
        return False
    return True       

我认为这个实现存在冗余问题：如果可迭代对象包含1000个同类型元素，会无意义地调用1000次type()函数。请问是否有更高效的实现方式？

优化实现方案 #

你可以提前终止遍历，不需要处理所有元素——只要发现第一个类型不一致的元素，就直接返回False，这样能大幅减少不必要的type()调用：

from typing import Iterable, Any

def has_one_datatype(it: Iterable[Any]) -> bool:
    iterator = iter(it)
    try:
        first_type = type(next(iterator))
    except StopIteration:
        # 空可迭代对象，可根据需求调整返回值，这里默认返回True
        return True
    for element in iterator:
        if type(element) != first_type:
            return False
    return True

优化说明 #

• 仅对第一个元素调用一次type()，后续元素只需与该类型做对比，避免重复调用
• 一旦发现类型不匹配的元素立即返回False，无需遍历剩余元素，在存在不同类型元素的场景下效率提升明显
• 处理了空可迭代对象的边界情况，你可以根据实际业务需求调整空对象的返回结果（比如认为空对象不符合"单一类型"要求，可改为返回False）

另外，你的原代码存在一个语法错误：Except需小写为except，优化代码中已修正该问题。

内容的提问来源于stack exchange，提问作者GarbageTime