【Python】使用列表推导式转换DF为Class.DF对象

最新推荐文章于 2025-03-07 22:29:07 发布

原创最新推荐文章于 2025-03-07 22:29:07 发布 · 428 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

Python 专栏收录该内容

9 篇文章

订阅专栏

文章介绍了一种方法，通过创建一个名为ClassA的类，其构造函数接受一个DataFrame对象。然后在ClassB中定义了一个方法group_and_convert_to_A_objects，该方法对DataFrame按C列进行分组，并将每个分组转换为ClassA的实例。这样做的目的是为了能够在ClassA中添加特定于DataFrame的操作，简化数据处理。

部署运行你感兴趣的模型镜像

现在有一个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简单。

您可能感兴趣的与本文相关的镜像