如何用Python pandas删除Excel中不含指定数字列表的行?
删除Excel中不包含指定数字的行
需求说明:需要删除Excel工作表中不包含指定数字列表中任一数字的行,目标列中的指定数字可能单独存在,也可能和其他数字、字符串混合在一起。
原代码问题分析
你提供的代码无法运行的原因是:
- pandas不支持直接将列与列表进行
!=的批量匹配 - 未考虑目标列中数字与其他内容混合的场景,直接做相等判断完全不符合需求
正确实现代码
以下是可以直接运行的解决方案,兼顾目标列是数字、字符串或混合类型的情况:
import pandas as pd # 读取Excel工作表,index_col=0根据你的实际需求决定是否保留 df = pd.read_excel('sheet.xlsx', index_col=0) # 定义需要保留的数字列表 target_numbers = [123456, 1234567, 12345678] # 将数字列表转为正则匹配模式,实现"包含任一数字"的匹配逻辑 match_pattern = '|'.join(map(str, target_numbers)) # 筛选出Column1中包含目标数字的行,直接覆盖原数据框 df = df[df['Column1'].astype(str).str.contains(match_pattern)] # 将处理后的结果保存到新Excel文件 df.to_excel('filtered_sheet.xlsx')
代码说明
astype(str):统一将目标列转为字符串类型,确保无论单元格是数字还是字符串,都能被正则匹配覆盖str.contains(match_pattern):检查每个单元格内容是否包含目标列表中的任一数字,返回布尔值用于筛选- 直接通过布尔索引筛选行,比
drop方法更直观高效
如果需要原地修改原数据框(不创建新变量),可以替换筛选行的代码为:
df.drop(df[~df['Column1'].astype(str).str.contains(match_pattern)].index, inplace=True)
内容的提问来源于stack exchange,提问作者Johntix
相关产品推荐
相关产品推荐

