python DataFrame筛除/删除不满足条件的行/数据

儒雅芝士

已于 2024-03-20 15:52:00 修改

阅读量413

点赞数 3

文章标签： python 开发语言

于 2023-11-23 16:35:17 首次发布

本文链接：https://blog.csdn.net/qq_42806827/article/details/134580894

版权

for遍历中，不满足条件的直接continue

如果for循环中有满足条件的，也有不满足条件的，需要drop剔除不满足条件的行

再保存

mainLaneData = pd.DataFrame()
for curid, curgroup in tracksdata.groupby("trackId"):
    print("遍历",curid)
    curgroup.sort_index(inplace=True)
    tmpconcat = curgroup
    tmpconcat["vehicleClass"] = tracksmetadata[tracksmetadata["trackId"] == curid]["class"].values[0]

    # 去掉truck类型的车辆，跳过for循环，没有执行数据合并
    VehicleClass = tmpconcat["vehicleClass"].unique()
    if VehicleClass == 'truck':
        continue

    # 不满足条件的行直接drop
    for index, row in curgroup.iterrows():
        if row["laneletId"] not in laneletID.lanlet2data["5"]["-2"]:
            tmpconcat.drop(index=index, inplace=True)
    # 合并数据
    mainLaneData = pd.concat([mainLaneData, tmpconcat], axis=0)