Python Union联合类型详解 - 多类型标注与类型窄化指南

from typing import Union

# 基本用法：值可以是多种类型之一
def func(param: Union[int, str]) -> None:
    ...

# 多种类型联合
def process(data: Union[int, float, str]) -> str:
    ...

# Union包含None（等价于Optional）
def maybe_func(param: Union[str, None]) -> None:
    ...

# Python 3.10+ 使用 | 运算符
def func(param: int | str) -> None:
    ...

from typing import Union

def process_value(value: Union[int, str]) -> str:
    """处理整数或字符串类型的值"""
    if isinstance(value, int):
        return f"整数: {value} 的平方是 {value ** 2}"
    else:
        return f"字符串: '{value}' 的长度是 {len(value)}"

# 传入整数
result1 = process_value(5)
print(result1)

# 传入字符串
result2 = process_value("hello")
print(result2)

整数: 5 的平方是 25
字符串: 'hello' 的长度是 5

from typing import Union, List

def describe(data: Union[int, float, str, List]) -> str:
    """描述不同类型的数据"""
    if isinstance(data, int):
        return f"这是一个整数: {data}"
    elif isinstance(data, float):
        return f"这是一个浮点数: {data:.2f}"
    elif isinstance(data, str):
        return f"这是一个字符串: '{data}'"
    elif isinstance(data, list):
        return f"这是一个列表，包含 {len(data)} 个元素"
    else:
        return "未知类型"

# 测试不同类型
print(describe(42))
print(describe(3.14159))
print(describe("Python"))
print(describe([1, 2, 3, 4, 5]))

这是一个整数: 42
这是一个浮点数: 3.14
这是一个字符串: 'Python'
这是一个列表，包含 5 个元素

from typing import Union, Optional, List, Dict

# 模拟API响应数据
ResponseData = Union[Dict[str, str], List[Dict[str, str]], str]

def parse_api_response(response: ResponseData) -> List[str]:
    """解析API响应，统一返回字符串列表"""
    result: List[str] = []

    if isinstance(response, str):
        # 纯文本响应
        result.append(response)
    elif isinstance(response, dict):
        # 单个对象响应
        for key, value in response.items():
            result.append(f"{key}: {value}")
    elif isinstance(response, list):
        # 列表响应
        for item in response:
            if isinstance(item, dict):
                result.append(str(item))
            else:
                result.append(item)

    return result

# 测试不同类型的响应
text_response = "操作成功"
dict_response = {"status": "ok", "message": "已更新"}
list_response = [
    {"id": "1", "name": "Alice"},
    {"id": "2", "name": "Bob"}
]

print("文本响应:", parse_api_response(text_response))
print("字典响应:", parse_api_response(dict_response))
print("列表响应:", parse_api_response(list_response))

文本响应: ['操作成功']
字典响应: ['status: ok', 'message: 已更新']
列表响应: ["{'id': '1', 'name': 'Alice'}", "{'id': '2', 'name': 'Bob'}"]