создать класс с кадром данных в качестве аргумента в python

Question

Я новичок в работе с Python и работал с объектной ориентацией только в небольшой степени в Java. Теперь я хочу создать класс с фреймом данных в качестве атрибута, который должен быть частью конструктора. Однако я не знаю, как этого добиться. Я попробовал следующее для класса (метод предназначен только для проверки работы кода)

import pandas as pd
data = pd.DataFrame()
class ProcessData():
    def __init__(self, frame):
        self.data = frame
    
    def time(self):
        return (self.data['time'])

и следующие для его использования

test = ProcessData(data)
print(test.time())

с данными, которые являются моим фреймом данных, который Я хочу передать колонку «время». Что я делаю не так?

Answer 1

Вы должны инициировать определение столбца «время» в вашей переменной данных. Например.

data = pd.DataFrame(columns = ['time'])
data['time'] = [1,2,3]