Python 将类方法应用于数据框的行

Question

我的类使用数据帧的一行来构造一个对象，我想通过将 init 应用于数据帧的每一行来创建一个对象数组。有没有办法对其进行矢量化？我的类定义如下:

class A(object):
    def __init__(self,row):
        self.a = row['a']
        self.b = row['b']

任何建议都将受到高度赞赏!

我有一种我不太满意的方法来解决这个问题。在类之外定义另一个函数，然后使用 apply。

def InitA(row):
    return A(row)

假设 df 是我想用作参数的数据框。

xxx = df.apply(InitA,axis=1)

给了我想要的。不过，我认为 InitA 不是必需的。

我原来的问题有点复杂。类定义为

class A(object):
    def __init__(self):
        return
    def add_parameter(self,row):
        self.a = row['a']

我打算将 add_parameter 应用于数据帧的每一行。但我认为定义另一个(lambda)函数对于解决这个问题是必要的。

Answer 1

只使用 lambda 函数？

xxx = df.apply(lambda x: A(x),axis=1)

编辑: 另一种解决方案是直接传递类，然后 apply-function 调用构造函数:

xxx = df.apply(A,axis=1)

这有效:

import pandas as pd 

class C(object):
    def __init__(self,dat):
        return

A = pd.DataFrame({'a':pd.Series([1,2,3])})
A.apply(lambda x: C(x),axis=1)