按子串位置筛选Pandas DataFrame行？解决类型错误

嘿，这个问题我之前也碰到过！出现TypeError: invalid type comparison的核心原因是你的KEY列里混有非字符串类型的数据（比如整数、浮点数，或者缺失值NaN），导致str[2:4]切片后得到的结果里存在无法和字符串"02"直接比较的类型，触发了类型不匹配错误。

下面是具体的解决步骤：

• 第一步：统一列类型为字符串
  先把KEY列强制转换成字符串类型，确保所有值都能正常进行字符串切片操作：

  df["KEY"] = df["KEY"].astype(str)
  

  如果你的列里有NaN缺失值，转换后会变成字符串"nan"，要是你想提前排除这些无效行，可以先执行过滤：

  # 先删除KEY列的缺失值行
  df = df.dropna(subset=["KEY"])
  # 再转换为字符串类型
  df["KEY"] = df["KEY"].astype(str)
  

• 第二步：执行筛选操作
  类型统一后，再运行你原本的筛选代码就可以正常工作了：

  df = df[df["KEY"].str[2:4] == "02"]
  

• 进阶：处理短字符串的情况
  要是有些KEY值的长度不足4位，str[2:4]会返回空字符串或者截断后的结果。如果你只想保留长度≥4位且子串符合条件的行，可以额外加个长度判断：

  df = df[(df["KEY"].str.len() >= 4) & (df["KEY"].str[2:4] == "02")]
  

简单解释下原理：Pandas的str访问器在处理非字符串值时，会返回NaN，而NaN和字符串"02"进行比较时，因为类型不兼容，就会抛出你遇到的那个错误。统一转换为字符串后，所有切片结果都是字符串类型，就能正常和"02"比较啦。

内容的提问来源于stack exchange，提问作者telex-wap