如何在函数中使用查询并应用于其他数据框？答案

【问题标题】：How to use query inside a function and apply to other dataframes?如何在函数中使用查询并应用于其他数据框？
【发布时间】：2020-10-18 14:17:31
【问题描述】：

我有一个数据框用户，并调用了公共列是 user_id 的地方。我需要在 churn 不为空的用户数据框中删除值并删除调用中的那些 user_id 行。

users = user_id,first_name,last_name,age,city,reg_date,plan,churn_date
1000,Anamaria,Bauer,45,"Atlanta-Sandy Springs-Roswell, GA MSA",2018-12-24,ultimate,
1001,Mickey,Wilkerson,28,"Seattle-Tacoma-Bellevue, WA MSA",2018-08-13,surf,
1002,Carlee,Hoffman,36,"Las Vegas-Henderson-Paradise, NV MSA",2018-10-21,surf,
1003,Reynaldo,Jenkins,52,"Tulsa, OK MSA",2018-01-28,surf,
1004,Leonila,Thompson,40,"Seattle-Tacoma-Bellevue, WA MSA",2018-05-23,surf,
1005,Livia,Shields,31,"Dallas-Fort Worth-Arlington, TX MSA",2018-11-29,surf,
1007,Eusebio,Welch,42,"Grand Rapids-Kentwood, MI MSA",2018-07-11,surf,
1008,Emely,Hoffman,53,"Orlando-Kissimmee-Sanford, FL MSA",2018-08-03,ultimate,
1009,Gerry,Little,19,"San Jose-Sunnyvale-Santa Clara, CA MSA",2018-04-22,surf,
1010,Wilber,Blair,52,"Dallas-Fort Worth-Arlington, TX MSA",2018-03-09,surf,

calls = id,user_id,call_date,duration
1000_93,1000,2018-12-27,8.52
1000_145,1000,2018-12-27,13.66
1000_247,1000,2018-12-27,14.48
1000_309,1000,2018-12-28,5.76
1000_380,1000,2018-12-30,4.22
1000_388,1000,2018-12-31,2.2
1000_510,1000,2018-12-27,5.75
1000_521,1000,2018-12-28,14.18
1000_530,1000,2018-12-28,5.77
1000_544,1000,2018-12-26,4.4

我想创建一个函数，以便在过滤掉原始用户数据帧后，我可以应用于其他数据帧，仅包含 user_id。我也想在其他数据框中删除其他 user_id

【问题讨论】：

标签： python pandas numpy data-science

【解决方案1】：

您可以尝试使用.query 进行简单过滤

def filter_calls_df(user_id_df, calls_df):

    # filter out churn user_id
    usr_id = user_id_df.query("churn_date.notna()")['user_id'].unique()

    # filter calls dataframe
    calls_df = calls_df.query("user_id not in @usr_id")
    
    return calls_df

【讨论】：

当函数使用两个参数时，如何将其应用于调用和其他数据帧？