如何在argparse中指定input=<INPUT>格式的命令行参数？

argparse默认只识别带-（短参数）或--（长参数）前缀的参数，直接把--input改成input=肯定不行，因为它不符合argparse的参数解析规则。要实现input=<INPUT>这种不带前缀的键值对形式，得自己处理参数解析逻辑，下面给两种可行方案：

方案一：手动提取目标参数，剩余参数交给argparse处理 #

这种方法可以保留argparse对其他参数的解析能力，同时单独处理input=格式的参数：

import argparse
import sys

input_value = None
remaining_args = []

# 遍历命令行参数，提取input的值
for arg in sys.argv[1:]:
    if arg.startswith('input='):
        input_value = arg.split('=', 1)[1]
    else:
        remaining_args.append(arg)

# 初始化argparse，处理其他参数（如果有）
parser = argparse.ArgumentParser()
# 这里可以添加其他需要的参数，比如：
# parser.add_argument('-o', '--output', help='Output file path')

# 解析剩余参数
args = parser.parse_args(remaining_args)

# 检查input参数是否存在
if not input_value:
    parser.error("必须提供input=<INPUT>参数")

# 后续使用input_value和args中的其他参数
print(f"输入值：{input_value}")
# 如果有output参数：print(f"输出路径：{args.output}")

使用时直接在命令行输入：

python your_script.py input=test.txt -o output.txt

方案二：将input=<INPUT>作为位置参数，验证格式 #

如果你的脚本只需要这一个参数，或者可以接受把input=<INPUT>当作唯一的位置参数，也可以用这种方式：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('input_arg', help="必须以input=<INPUT>格式传入")
args = parser.parse_args()

# 验证参数格式并提取值
if not args.input_arg.startswith('input=') or '=' not in args.input_arg:
    parser.error("参数格式错误，请使用input=<INPUT>形式")

input_value = args.input_arg.split('=', 1)[1]
print(f"输入值：{input_value}")

使用时输入：

python your_script.py input=test.txt

说明 #

argparse的设计初衷是处理带前缀的命令行参数，对于不带前缀的key=value形式，它会默认当作位置参数处理，所以必须手动介入解析逻辑。如果需要支持多个类似key=value的无前缀参数，可以扩展方案一的逻辑，遍历提取所有符合格式的键值对。

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