使用GroupBy apply时保留NaN组

import pandas as pd

# 创建DataFrame示例数据
data = {'A': [1, 2, 3, 4, 5],
        'B': [5, 4, 3, 2, 1],
        'C': [1, 2, None, 4, 5]}
df = pd.DataFrame(data)

# 按列A进行分组，并对每个组应用自定义函数
def custom_func(group):
    # 在自定义函数中处理NaN组
    if pd.isna(group['C']).any():
        # 处理NaN组的逻辑
        pass
    else:
        # 处理非NaN组的逻辑
        pass
    return group

# 使用GroupBy apply时保留NaN组
result = df.groupby('A', dropna=False).apply(custom_func)