Pandas 使用自定义函数进行变换

Question

这是我的简单代码

def custom_func(x):
    for el in x.index:
        print(el)
    return None

df = pd.DataFrame({'A':['F','F','F','K','K'], 'B':[54,87,35,25,82],
                       'C':[56,78,0,14,13]})

mask = df.groupby(['A'])['B'].transform(custom_func)

它会生成以下错误消息：

Traceback (most recent call last):
  File "C:/Users/temp3.py", line 59, in <module>
    consecutive_check()
  File "C:/Users/temp3.py", line 56, in consecutive_check
    mask = df.groupby(['A'])['B'].transform(custom_func)
  File "C:\Anaconda3\lib\site-packages\pandas\core\groupby.py", line 2763, in transform
    result[indexer] = res
TypeError: int() argument must be a string, a bytes-like object or a number, not 'NoneType'

怎么了，为什么我会收到这个错误信息。

Answer 1

也许是因为你的函数总是返回None，不管输入是什么。您能否更完整地描述您期望的最终结果？