如何在Python函数中使用可选命令行参数并优化条件判断逻辑？

嘿，手动判断sys.argv的长度确实不够优雅，而且参数越多，条件分支会越乱。这里给你推荐两种改进方案，优先推荐用Python标准库的argparse——它是处理命令行参数的官方工具，简洁、健壮还自带帮助功能。

方案一：使用argparse模块（推荐） #

这个模块可以帮你自动处理参数解析、类型转换、默认值，还能生成友好的帮助信息，扩展性拉满。改写你的代码如下：

import argparse

def getRelevant(number=5, sortBy="Change"):
    # 保留你的原有业务逻辑
    relevant = f"获取前{number}条数据，按{sortBy}字段排序"
    return relevant

if __name__ == "__main__":
    # 创建参数解析器，添加描述信息
    parser = argparse.ArgumentParser(description="用于获取相关数据的命令行工具")
    
    # 添加可选参数：设置短选项、长选项、类型、默认值和帮助文本
    parser.add_argument("-n", "--number", type=int, default=5, 
                        help="要获取的条目数量，默认值为5")
    parser.add_argument("-s", "--sort-by", default="Change", 
                        help="排序依据的字段，默认值为Change")
    
    # 解析命令行传入的参数
    args = parser.parse_args()
    
    # 调用函数并打印结果
    print(getRelevant(args.number, args.sortBy))

为什么这个方案更好？ #

• 无需手动判断参数数量：argparse会自动处理用户传0个、1个或2个参数的情况，完全替代原来的if-elif-else逻辑。
• 自动类型转换：--number参数指定了type=int，用户输入的字符串会自动转成整数，不用你手动写int(number)做类型转换。
• 自带帮助文档：用户运行python your_script.py -h就能看到清晰的参数说明，体验友好：

  usage: your_script.py [-h] [-n NUMBER] [-s SORT_BY]
  
  用于获取相关数据的命令行工具
  
  optional arguments:
    -h, --help            show this help message and exit
    -n NUMBER, --number NUMBER
                          要获取的条目数量，默认值为5
    -s SORT_BY, --sort-by SORT_BY
                          排序依据的字段，默认值为Change
  

• 扩展性强：以后要加新参数，只需要多一行parser.add_argument()，不用修改任何条件判断逻辑。

方案二：轻量版（适合简单场景） #

如果你的需求非常简单，也可以用sys.argv切片结合函数的默认参数来简化代码：

import sys

def getRelevant(number=5, sortBy="Change"):
    number = int(number)
    relevant = f"获取前{number}条数据，按{sortBy}字段排序"
    return relevant

if __name__ == "__main__":
    # 把sys.argv[1:]的元素自动解包传递给函数，匹配默认参数逻辑
    print(getRelevant(*sys.argv[1:]))

注意事项 #

• 这个方式虽然简洁，但没有类型检查：如果用户给number传了非整数，会直接触发类型转换错误。
• 没有帮助提示，用户不知道该传什么参数、参数顺序是什么。
• 参数顺序固定：必须先传number再传sortBy，不像argparse可以通过选项名指定任意顺序。

所以如果你的脚本需要给他人使用，或者未来可能新增参数，优先选择argparse方案。

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