如何将lambda函数映射到可为float、int或list类型的变量？

问题根源：map()函数要求第二个参数是可迭代对象，而float/int属于不可迭代类型，直接传入会抛出TypeError；且list(val)无法将单个数值转为可迭代对象（因为数值本身不可迭代）。

以下是几种兼容两种场景的实现方式：

方案1：类型判断包装（最直观） #

先写一个辅助函数，将单个数值包装为列表，列表类型直接返回：

def ensure_iterable(val):
    if isinstance(val, (int, float)):
        return [val]
    return val

# 测试单个float
val = 0.1
test = list(map(lambda x: max(x, 1 - x), ensure_iterable(val)))
print(test)  # 输出: [0.9]

# 测试列表
val = [0.1, 0.6]
test = list(map(lambda x: max(x, 1 - x), ensure_iterable(val)))
print(test)  # 输出: [0.9, 0.6]

方案2：三元表达式简化（无需辅助函数） #

直接在调用map时用三元表达式处理输入：

# 单个数值场景
val = 0.1
test = list(map(lambda x: max(x, 1 - x), [val] if isinstance(val, (int, float)) else val))

# 列表场景
val = [0.1, 0.6]
test = list(map(lambda x: max(x, 1 - x), [val] if isinstance(val, (int, float)) else val))

方案3：通用兼容（含异常捕获） #

如果需要兼容更多非可迭代类型（排除字符串，避免误处理），可以用try-except实现：

def ensure_iterable(val):
    try:
        iter(val)
        if isinstance(val, str):
            return [val]
        return val
    except TypeError:
        return [val]

# 测试单个int
val = 3
test = list(map(lambda x: max(x, 1 - x), ensure_iterable(val)))
print(test)  # 输出: [3]

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