如何在Python中实现单行单输入或多输入的兼容处理

你可以通过以下几种方式实现让用户自主输入1个或2个值，避免触发ValueError：

方法1：先拆分再根据长度赋值 #

先将输入拆分为列表，再判断列表长度来给变量赋值，这种方式直观易懂：

# 获取拆分后的输入列表
input_parts = input().split()
# 第一个值必取
a = input_parts[0]
# 第二个值存在则取，否则设为占位符（这里用None，可替换为''、0等）
b = input_parts[1] if len(input_parts) >= 2 else None

方法2：利用解包语法简化代码 #

Python的扩展解包语法可以更简洁地处理这种场景：

# 用*rest接收除第一个值外的所有输入
a, *rest = input().split()
# 若rest不为空则取第一个元素作为b，否则用占位符
b = rest[0] if rest else None

如果需要限制最多只能输入2个值，可以先对拆分后的列表切片：

input_parts = input().split()[:2]
# 补全占位符后解包，确保总能分到2个值
a, b = input_parts + [None] * (2 - len(input_parts))

方法3：用异常捕获处理（适合需要区分输入场景的情况） #

如果需要在用户输入不符合预期时做额外处理，可以捕获ValueError：

try:
    a, b = input().split()
except ValueError:
    # 仅输入1个值时的处理逻辑，避免重复获取输入
    a = input().split()[0]
    b = None

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