我正在使用 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
提交
取消