Python中处理可变数量键盘输入，避免ValueError的方法

嘿，这个问题我之前写数据处理脚本时也踩过坑！核心原因很简单：当你用country1, country2, country3 = ...这种方式解包时，Python要求等号右边的可迭代对象（这里是split(',')返回的列表）必须正好有3个元素，但输入1个或2个国家时，列表长度不够，就会抛出ValueError。

下面给你几个实用的解决方案，你可以根据自己的需求选：

方案1：补全列表长度后解包（推荐，兼顾鲁棒性） #

这个方法会先处理输入的空格、空值，再把列表补全到3个元素，不足的用空字符串填充，同时限制最多取3个国家：

# 获取输入并预处理：去掉前后空格，分割后去掉每个元素的空格，过滤空字符串
country_input = input("Write up to three comma-separated countries for which you want to extract data: ").strip()
countries = [c.strip() for c in country_input.split(',') if c.strip()]

# 最多取3个国家，不足的补空字符串
countries = countries[:3] + [''] * (3 - len(countries))

# 安全解包
country1, country2, country3 = countries

这样不管用户输入1个、2个、3个，甚至输入带空格（比如"Canada, Mexico, Brazil"）或者多打了逗号（比如"Italy,,Spain"），都能正常处理。

方案2：用扩展解包语法（简洁） #

Python 3支持扩展解包，用*来处理多余或不足的元素，这里我们可以先把列表补全到至少3个元素，再解包：

country_input = input("Write up to three comma-separated countries for which you want to extract data: ").strip()
countries = [c.strip() for c in country_input.split(',') if c.strip()]

# 补全到3个元素，多余的用*_忽略
country1, country2, country3, *_ = countries + [None] * 3

这里没输入的国家会被赋值为None，你也可以把None换成''（空字符串），根据自己的业务逻辑调整。

方案3：逐个赋值（直观，适合新手） #

如果觉得上面的方法有点绕，也可以用最直白的方式，先初始化变量，再根据输入列表的长度赋值：

country_input = input("Write up to three comma-separated countries for which you want to extract data: ").strip()
countries = [c.strip() for c in country_input.split(',') if c.strip()]

# 初始化默认值
country1 = country2 = country3 = ''

# 根据输入数量赋值
if len(countries) >= 1:
    country1 = countries[0]
if len(countries) >= 2:
    country2 = countries[1]
if len(countries) >= 3:
    country3 = countries[2]

这个逻辑非常清晰，容易理解和调试。

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