我正在使用 python3 和 pandas,我想:作为函数参数传递,以声明传递给df.loc.例如,假设我有一个像这样填充 na 值的函数:def fill_na_w_value(df, rows, columns, fill): for col in columns: df.loc[rows, columns].fillna( fill, inplace=True )有时我可能不想将其应用于某些行,而是将其应用于所有行,在 Pandas 中,这是通过访问df.loc[:, col]如果我从一个函数调用它,它会喜欢fill_na_w_value(df, :, ['col1'], 0)但是上面会给我一个语法错误,因为:; 如何将其作为函数参数传递?
1 回答
达令说
TA贡献1821条经验 获得超6个赞
使用slice(None)来表示:。请注意,您可以使用pipe通过函数传递数据框,并loc接受用于行和索引过滤的列表:
df = pd.DataFrame({'col1': [1, 2, np.nan, 4, 5, np.nan, 7, 8, np.nan]})
def fill_na_w_value(df, row_slicer, columns, value):
df.loc[row_slicer, columns] = df.loc[row_slicer, columns].fillna(value)
return df
df1 = df.pipe(fill_na_w_value, slice(None), ['col1'], 0)
print(df1)
col1
0 1.0
1 2.0
2 0.0
3 4.0
4 5.0
5 0.0
6 7.0
7 8.0
8 0.0
这是一个使用 alist而不是slice对象的示例:
df2 = df.pipe(fill_na_w_value, [2, 5], ['col1'], 0)
print(df2)
col1
0 1.0
1 2.0
2 0.0
3 4.0
4 5.0
5 0.0
6 7.0
7 8.0
8 NaN
添加回答
举报
0/150
提交
取消