现在有一个Class,里面有一个df, 我现在要把A的df按照列C分组,
最终结果的每个元素都转换成类的df对象。
做法如下:
class A:
def __init__(self, df):
self.df = df
class B:
def group_and_convert_to_A_objects():
grouped_data = df.groupby('C')
# 使用列表推导式来将每个分组数据转换为A.df对象
list_a_obj = [A(group_df) for _, group_df in grouped_data]
return list_a_obj
这样做的好处是,可以在A类里加一些针对DF的方法。
有些时候会比直接处理DF简单。