python - 如何使用已准备好列/列参数列表的功能扩展熊猫？

Question

有一个不错的库 pandas-flavor https://pypi.org/project/pandas-flavor/允许扩展 pandas。

我写了这样的玩具示例：

import pandas as pd
import pandas_flavor as pf
from scipy import stats


df = pd.DataFrame(data={
  "x": ['a', 'a', 'b'],
  "y": [0, 2, 5],
  "z": [5, 0, 7],
})


@pf.register_dataframe_method
def to_frame(df):
  return df

@pf.register_dataframe_method
def custom_func(df, grouping, c):
  return df.groupby(grouping)[c].transform(stats.zscore).to_frame().max(axis=1)

基本上，c参数现在可以是单个列（字符串）或custom_func（主要目标）的列列表：

df.custom_func(['x'], 'y')
df.custom_func(['x'], ['y'])
df.custom_func(['x'], ['y', 'z'])

几点说明：

• 对于单列df.groupby(grouping)[c].transform(stats.zscore)返回Series，它max(axis=1)会给出错误，除非你把它变成DataFrame( to_frame())
• 对于多个列df.groupby(grouping)[c].transform(stats.zscore)给出DataFrame但没有to_frame()它，所以我通过 pandas-flavor 添加了它

它有效，但问题是：

• 可以更简单或更标准的方式完成吗？
• 是否有标准方法来编写可用于单列/多列的函数？

标签： pythonpandaspandas-groupbycustomization