如何使用pandas拆分数据集生成子表并删除含-99的行
问题修正方案
你原有代码的核心问题有两个:
- 生成PCP列名时
range(1,n_cols)是左闭右开区间,当n_cols=4时只会生成1/2/3三个值,漏掉了PCP4列 - 最后过滤行的循环逻辑错误:混淆了字典的key和实际存储的DataFrame对象,同时判断相等的运算符用错,且筛选条件应该是保留不等于-99的行
最简实现方案
你可以把生成子DataFrame和过滤无效行的步骤合并,减少冗余循环:
import pandas as pd # 假设你已经读取原始数据集并存为变量df n_cols = 4 dfs = {} for i in range(1, n_cols + 1): pcp_col = f"PCP{i}" # 取指定列 + 过滤-99行 + 重置索引 sub_df = df[["date", pcp_col]][df[pcp_col] != -99].reset_index(drop=True) dfs[f"df{i}"] = sub_df # 调用示例:查看df2的内容直接用 dfs["df2"]
兼容原有写法的修正方案
如果你要保留先生成所有子DataFrame再统一过滤的逻辑,把最后一段循环替换为以下代码即可:
for dfn, pcp_col in zip(df_names, pcp_list): # 筛选当前子DF中PCP列不等于-99的行,重新赋值给字典 dfs[dfn] = dfs[dfn][dfs[dfn][pcp_col] != -99].reset_index(drop=True)
内容的提问来源于stack exchange,提问作者Naru
相关产品推荐
相关产品推荐

