Python函数返回可变数量输出：是否有更简洁的f2实现方式？

基础实现 #

def f(*args):
    return args

print(f(1))
print(f())

# 输出
# (1,)
# ()

大致符合需求的实现 #

def f2(*args):
    if len(args) > 1:
        return args
    elif len(args) == 1:
        return args[0]
    else:
        return None
    
print(f2(1))
print(f2())

# 输出
# 1
# None

技术问询 #

1. 是否有更简洁的方式实现f2函数？
2. 常规return语句可直接返回单个值，但处理参数化输入时却需专门将元组转换为单个值，这似乎有些不一致。

更简洁的实现方式 #

可以用条件表达式简化代码，一行就能完成需求：

def f2(*args):
    return args if len(args) > 1 else args[0] if args else None

也可以利用元组的布尔特性写出更直观的版本：

def f2(*args):
    return args[0] if len(args) == 1 else args or None

两种写法都和原f2函数逻辑完全一致，代码更紧凑。

关于“不一致”的说明 #

这种差异是*args的特性导致的：

• 直接return x时，x本身就是单个值，自然返回单个值；
• *args的作用是把传入的位置参数打包成元组，无论传0个、1个还是多个参数，它本质都是元组类型。

要实现“传一个返回单个值，传多个返回元组，传0个返回None”的逻辑，必须对这个元组做判断转换——这不是语法设计的不一致，而是*args的打包特性和你的需求匹配问题。如果想避免转换，要么不使用*args（直接定义固定参数），要么接受返回值始终是元组（如基础实现）。

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