dataframe 根据条件获取元素值

---

应用场景

有下面这样的一张dataframe表格

有如下需求：
给定业务运营和工作类型的值，能获取工程范围的值。
若未查询到，则为空

解决代码

def get_works_scope(input_df,service_operation,worktype):
    '''根据“业务运营”和 “工作类型”获取“工程范围”的值
    Args:
        input_df (df): 输入的Dataframe 表格
        service_operation (string): 业务运营
        worktype (string): 工作类型_

    Returns:
        _type_: _description_
    '''
    out_serise=input_df.iloc[(input_df['业务运营']==service_operation) & (input_df['工作类型']==worktype)]#根据条件可以获取一个serise
    if (len(out_serise)==0):#out_serise为空，未查询到
        return np.nan
    else :
        return  out_serise['工程范围'].loc[0] #不为空返回，数值

结果演示

样例一

查询 业务运营为‘业务运营1’，工作类型为‘性能优化’ 的工程范围值

与表格对应一致

样例二

查询 业务运营为‘业务运营2’，工作类型为‘性能优化’ 的工程范围值

与表格对应一致