使用Python遍历CSV列检测字符串存在性时的匹配异常问题

你的代码逻辑存在核心问题：遍历Target列时，只要当前item和输入不匹配就立刻让用户重新输入，这会导致：

1. 第一个元素不匹配时，直接覆盖gene1的值，后续循环用新输入的值和下一个元素比对，完全没机会检查完所有元素
2. 就算某个元素匹配成功，后续循环的其他元素还是会触发else分支，再次让用户输入

修正方案 #

步骤1：优化查找逻辑 #

先把Target列的所有值提取为一个集合（集合的成员查找效率远高于遍历Series），然后循环接收用户输入，直到输入的值存在于集合中。

步骤2：处理潜在的空格问题 #

很多时候匹配失败是因为输入或CSV中的字符串前后有空格，所以要对输入和CSV的值做去空格处理。

修正后的代码 #

import pandas as pd

# 读取CSV并处理Target列的空格
rawdata = pd.read_csv("pcr.csv")
# 提取Target列的值，去空格后转为集合
target_set = set(rawdata["Target"].str.strip())

# 循环接收输入直到匹配成功
while True:
    gene1 = input("Enter a gene: ").strip()
    if gene1 in target_set:
        print(f"Gene 1: {gene1}")
        break
    else:
        print("Not found. Enter again: ")

关键改进点 #

• 用set存储Target列的值，in操作的时间复杂度从O(n)降到O(1)，查找更快
• 用while True循环替代遍历，确保用户输入会被完整检查所有Target值
• 用str.strip()处理CSV中的值和用户输入，避免空格导致的匹配失败
• 找到匹配值后用break退出循环，不会重复触发输入提示

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