Python如何实现城市、邮编、州任选其一的OR输入查询功能

原代码问题 #

原写法存在两个核心错误：

• Python语法不支持在or逻辑运算链中直接串联多个赋值语句，=赋值运算符优先级低于逻辑运算符，运行会直接报语法错误
• 没有做输入有效性校验，也没有对应DataFrame的条件查询逻辑，无法实现“任意填一项即可查询”的需求

正确实现代码 #

实现逻辑：依次提示用户输入三个字段，未掌握的信息可直接回车留空，校验到至少一项有有效输入后，根据已填写的字段构造组合查询条件，从DataFrame中筛选匹配的门店，返回距离最近的结果。

import pandas as pd

# --------------- 输入收集模块 ---------------
while True:
    print("提示：城市、邮编、州三项中任意填写一项即可查询，其余项可直接回车留空")
    enter_city = input("Enter City: ").strip()
    enter_zip_code = input("Enter Zip Code: ").strip()
    enter_state = input("Enter State: ").strip()

    # 校验是否存在有效输入
    if enter_city or enter_zip_code or enter_state:
        break
    print("错误：三项信息不能全为空，请至少填写一项\n")

# --------------- DataFrame查询模块 ---------------
# 替换为你实际的DataFrame变量名，以及对应字段名
# 示例df字段说明：city-城市, zip_code-邮编, state-州, distance-用户到门店的距离, location-门店地址
query_mask = [True] * len(store_df)

if enter_city:
    # 统一转小写匹配，忽略大小写差异
    query_mask = query_mask & (store_df["city"].str.lower() == enter_city.lower())
if enter_zip_code:
    # *注意：邮编如果存为数字类型，必须先转字符串再比对，避免类型不匹配报错*
    query_mask = query_mask & (store_df["zip_code"].astype(str) == enter_zip_code)
if enter_state:
    query_mask = query_mask & (store_df["state"].str.lower() == enter_state.lower())

matched_stores = store_df[query_mask]

if matched_stores.empty:
    print("未查询到符合条件的纸杯蛋糕门店，请核对输入信息后重试")
else:
    # 按距离升序排序，取第一条就是最近的门店
    nearest = matched_stores.sort_values(by="distance", ascending=True).iloc[0]
    print(f"Nearest Cupcake Location: {nearest['location']}")

可选调整说明 #

• 如果不需要兼容用户同时填写多个字段的场景，可以先加一步输入类型选择，让用户先选择要输入城市/邮编/州中的哪一种，再接收对应内容即可，逻辑更简单
• 字符串匹配前统一做strip()去除前后空格、转小写，是为了避免用户输入的格式问题导致匹配失败，可根据你的数据格式调整匹配规则
• 如果没有存储门店到用户位置的距离字段，可根据实际业务逻辑调整“最近门店”的筛选规则

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