python - 为 ":"切片定义一个变量

Question

我想定义一个函数，如果设置了另一个变量，则该函数将对数据帧的一部分进行操作，或者在其他情况下对整个数据帧进行操作。例如 :

def mean_loc(df, col_name, idx=None):
    if idx is not None:
        return df.loc[idx, col_name].mean()
    else:
        return df[col_name].mean()

这工作得很好，但我想知道这是否可以做这样的事情:

def mean_loc_2(df, col_name, idx=None):
    if idx is None:
        idx = :
    return df.loc[idx, col_name].mean()

这当然不起作用，但我想知道是否存在旁路？如果我这样做 idx = df.index ，评价 df.loc[df.index, col_name]比做要长得多 df.loc[:, col_name] .我试过使用 slice从内置函数做 idx = slice(len(df))但没有任何成功(它也更长并且不适用于 DateTime 索引)。

有谁知道如何做这样的事情？

Answer 1

:不是一个值；它是 Python 语法的一部分。您要的是slice : 的对象语法表示。

if idx is None:
    idx = slice(None)  # sadly, slice() does not work

自 slice object 是不可变的，使用 1 作为默认参数值是安全的。

def mean_loc_2(df, col_name, idx=slice(None)):
    return df.loc[idx, col_name].mean()