使用自定义数据类方法操作 DataFrame

Question

我有超过 4000 行代码用于分析、操作、比较和绘制 2 个巨大的 .csv 文档。为了可读性和将来的发布，我想转换为面向对象的类。我将它们转换为pd.DataFrames：

my_data1 = pd.DataFrame(np.random.randn(100, 9), columns=list('123456789'))
my_data2 = pd.DataFrame(np.random.randn(100, 4), columns=list('ABCD'))

我有比较每个数据集的各个方面的函数和只单独使用数据集的函数。我想将此结构转换为具有每个数据帧的方法的数据类。

我无法通过我的类函数操作这些数据帧。我不断收到NameError: name 'self' is not defined。这是我的数据类结构：

@dataclass
class Data:
    ser = pd.DataFrame 

    # def __post_init__(self):
    #     self.ser = self.clean()

    def clean(self, ser):
        acceptcols = np.where(ser.loc[0, :] == '2')[0]
        data = ser.iloc[:, np.insert(acceptcols, 0, 0)]
        data = ser.drop(0)
        data = ser.rename(columns={'': 'Time(s)'})
        data = ser.astype(float)
        data = ser.reset_index(drop=True)
        data.columns = [column.replace('1', '')
                        for column in ser.columns]

        return data


my_data1 = pd.DataFrame(np.random.randn(100, 9), columns=list('123456789'))
my_data2 = pd.DataFrame(np.random.randn(100, 4), columns=list('ABCD'))

# Attempt 1
new_data1 = Data.clean(my_data1) # Parameter "ser" unfilled 
# Attempt 2
new_data1 = Data.clean(ser=my_data1) # Parameter "self" unfilled 
# Attempt 3
new_data1 = Data.clean(self, my_data1) # Unresolved reference "self"

我尝试过各种形式的定义def clean(self and other stuff)，但我认为我对类或类结构还不够了解。关于类和数据类的文档总是使用非常基本的示例，我尝试剪切/粘贴模板无济于事。我错过了什么？

Answer 1

你可以先获取Data类的实例x。

x = Data()

# Attempt 1
new_data1 = x.clean(my_data1) # Parameter "ser" unfilled 
# Attempt 2
new_data1 = x.clean(ser=my_data1) # Parameter "self" unfilled 

如果我是你，我不会以这种方式使用类，而是只定义以下函数

def clean(ser):
        acceptcols = np.where(ser.loc[0, :] == '2')[0]
        data = ser.iloc[:, np.insert(acceptcols, 0, 0)]
        data = ser.drop(0)
        data = ser.rename(columns={'': 'Time(s)'})
        data = ser.astype(float)
        data = ser.reset_index(drop=True)
        data.columns = [column.replace('1', '')
                        for column in ser.columns]

        return data

并直接调用它。

另外，在您的clean() 中，每次修改都基于ser，这是输入，但不是最后一次修改。这是个问题，不是吗？