如何让Python函数支持同时接受整数或整数元组参数？

当然有办法啦！Python里有不少实用技巧能让你的函数同时兼容单个整数和整数元组参数，我给你分享几种常用方案：

1. 函数内部直接做参数适配（最直观） #

这是最简单粗暴的方式：在函数开头判断参数类型，如果是单个整数，就把它转换成长度为1的元组，这样后续逻辑就可以统一按元组来处理，不用改原有代码。

示例代码：

def process_ports(ports):
    # 把单个整数转为单元素元组
    if isinstance(ports, int):
        ports = (ports,)
    # 后续统一处理元组逻辑
    for port in ports:
        print(f"正在处理端口: {port}")

# 测试调用
process_ports(80)          # 传入单个整数，正常工作
process_ports((80, 443))   # 传入整数元组，同样正常工作

这种方式的好处是改动极小，原有处理元组的逻辑完全不用动，快速实现兼容。

2. 用functools.singledispatch实现类型专属逻辑（更优雅） #

如果你希望针对不同参数类型写独立的处理逻辑，可以用Python标准库的functools.singledispatch装饰器，实现类似静态语言的函数重载效果。

示例代码：

from functools import singledispatch

@singledispatch
def process_ports(ports):
    # 默认处理逻辑，遇到不支持的类型就报错
    raise TypeError("参数必须是整数或整数元组")

@process_ports.register(int)
def _(single_port):
    # 专门处理单个整数的逻辑
    print(f"处理单个端口: {single_port}")

@process_ports.register(tuple)
def _(port_tuple):
    # 专门处理整数元组的逻辑，还可以加参数校验
    for port in port_tuple:
        if not isinstance(port, int):
            raise ValueError("元组内所有元素必须是整数")
        print(f"处理端口: {port}")

# 测试调用
process_ports(80)          # 触发整数专属处理函数
process_ports((80, 443))   # 触发元组专属处理函数

这种方式更模块化，不同类型的逻辑分开维护，后续还能轻松扩展支持其他类型（比如列表）。

3. 接受可变参数（*args）（调用更灵活） #

另一种思路是让函数直接接受任意数量的整数参数，这样用户既可以传单个整数，也可以传多个整数（自动打包成元组），甚至可以解包已有的元组传入。

示例代码：

def process_ports(*ports):
    # 先检查至少传入一个参数
    if not ports:
        raise ValueError("请至少传入一个端口号")
    # 校验每个参数都是整数
    for port in ports:
        if not isinstance(port, int):
            raise TypeError("所有端口号必须是整数")
        print(f"处理端口: {port}")

# 测试调用
process_ports(80)          # 单个整数
process_ports(80, 443)     # 多个整数，自动打包成元组
process_ports(*(80, 443))  # 解包已有的元组传入

这种方式对用户更友好，调用时不用刻意打包成元组，体验更顺畅。

选择建议 #

• 如果只是简单兼容单个整数，第一种方案最快最省心；
• 如果需要针对不同类型写不同逻辑，选singledispatch更优雅；
• 如果想让函数调用更灵活，可变参数的方式体验最好。

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