python - 我如何通过: as a function parameter in python?

Question

我正在使用 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)

但是由于 :; ，上面的代码会给我一个语法错误。我如何将其作为函数参数传递？

Answer 1

使用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

这是一个使用列表而不是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