将统计代码添加到 Python 中的函数

Question

我对 Python 比较陌生，正在尝试学习如何编写函数。 this post 的答案强调了如何从数据框中获取某些统计信息，我想在函数中使用它。

这是我的尝试，但不适用于AttributeError: 'SeriesGroupBy' object has no attribute 'test_for_B'：

 def test_multi_match(df_in,test_val):
    test_for_B = df_in == test_val
    contigious_groups = ((df_in == test_val) & (df_in != df_in.shift())).cumsum() + 1
    counts = df_in.groupby(contigious_groups).test_for_B.sum()
    counts.value_counts() / contigious_groups.max()

有人可以帮忙把这段代码放在一个我可以在其他数据帧上重复使用的函数中吗？谢谢。

编辑：现在已经解决了大属性错误。

Answer 1

给你：

def repeat_stats(series, var):
    isvar = series == var
    wasntvar = series != series.shift()
    cont_grps = (isvar & wasntvar).cumsum()
    counts = isvar.loc[cont_grps.astype(bool)].groupby(cont_grps).sum()
    return counts.value_counts() / cont_grps.max()

repeat_stats(rng.initial_data, 'B')

3.0    0.5
2.0    0.5
Name: initial_data, dtype: float64