python - 有没有办法让创建的类从熊猫数据框中提取列

Question

我正在尝试通过创建一个类来为我创建一个现有的 pandas 数据框，当我运行该类时，它说 DataFrame 没有属性。这是我的代码

import pandas as pd

def csv_reader(filename):
    global df
    df = pd.read_csv(filename)
    return df
    
class frameEditor:
    def __init__(self, df):
        self.df = df
        
    def find_created_at(self):
        created_at = df['created_at'].tolist()
        return created_at
    
    def find_source(self):
        source = df['source'].tolist()
        return source
    
    def find_author(self):
        author = df['original_author'].tolist()
        return author
    
    def to_dataframe(self):
        columns = ['created', 'source', 'author']
        
        created = self.find_created_at()
        source = self.find_source()
        author = self.find_author()
        
        new_df = pd.DataFrame(columns)
        
        return new_df

标签： pythonpandasoop