Python中实现非整数输入判断及修复代码分支逻辑问题

问题场景 #

你编写的代码用于读取文本文件、统计字符总数，并根据输入的索引输出对应单词，但存在输入兼容性问题：

def main():
    filename = input("Enter filename: \n")
    print(f"The total number of characters in the file {filename} is: {character_count(filename)} \n")
    while True:
        n = int(input("\nEnter a index number that you want to check in the file or type 'S' to stop: "))
        if n == 'S':
            break
        else:
            print(f"\nThe index of {n} in the file is {find_word(filename, n)}")

• 输入整数时功能正常，但输入退出指令S会因int()强制转换抛出ValueError（非整数无法转成int）
• 若移除int()转换，索引输出功能失效，无法将输入传递给find_word函数作为有效索引

解决方案 #

核心思路是先保留原始输入字符串，再分别处理退出指令和整数转换，以下提供两种可行实现：

方法1：异常捕获法（推荐）

利用try-except捕获整数转换失败的异常，同时优先判断退出指令：

def main():
    filename = input("Enter filename: \n")
    print(f"The total number of characters in the file {filename} is: {character_count(filename)} \n")
    while True:
        user_input = input("\nEnter a index number that you want to check in the file or type 'S' to stop: ").strip()
        # 优先处理退出指令，支持大小写输入
        if user_input.upper() == 'S':
            break
        # 尝试转换为整数，失败则提示无效输入
        try:
            n = int(user_input)
            print(f"\nThe index of {n} in the file is {find_word(filename, n)}")
        except ValueError:
            print("无效输入，请输入整数或'S'退出")
            # 若希望无效输入直接退出，可替换为 break

• 用strip()去除输入前后空格，避免带空格的S被误判
• upper()让输入s也能触发退出逻辑，提升用户体验
• 捕获ValueError处理所有非整数的无效输入，逻辑简洁全面

方法2：字符串校验法

先校验输入是否为整数格式，再进行转换：

def main():
    filename = input("Enter filename: \n")
    print(f"The total number of characters in the file {filename} is: {character_count(filename)} \n")
    while True:
        user_input = input("\nEnter a index number that you want to check in the file or type 'S' to stop: ").strip()
        if user_input.upper() == 'S':
            break
        # 校验是否为整数（兼容负索引）
        if user_input.lstrip('-').isdigit():
            n = int(user_input)
            print(f"\nThe index of {n} in the file is {find_word(filename, n)}")
        else:
            print("无效输入，请输入整数或'S'退出")

• lstrip('-')去除负号后用isdigit()判断是否为数字字符串，支持负索引场景
• 逻辑直观，但仅能处理纯数字格式的整数输入，无法处理带空格的数字（如" 123 "已被strip()处理）

内容的提问来源于stack exchange，提问作者Monkey D Luffy