Python 按行在数据帧上应用函数

Question

我不明白当函数有其他参数时，在创建函数时应如何使用“行”参数。我想创建一个函数来计算我的数据框"file"的新列。

这很好用:

def imputation(row):    
    if  (row['hour_y']==0) & (row['outlier_idx']==True) :
        val=file['HYDRO'].mean()
    else : 
        val=row['HYDRO']
    return val

file['minute_corr'] = file.apply(imputation, axis=1) 

但这不起作用(我添加了一个参数):

def imputation(row,variable):    
    if  (row['hour_y']==0) & (row['outlier_idx']==True) :
        val=file[variable].mean()
    else : 
        val=row[variable]
    return val
file['minute_corr'] = file.apply(imputation(,'HYDRO'), axis=1) 

Answer 1

尝试这种矢量化方法:

file['minute_corr'] = np.where((file['hour_y']==0) & file['outlier_idx'],
                               file['HYDRO'].mean(),
                               file['HYDRO'])