Вызов pandas фрейма данных в функции путем указания ската - PullRequest
0 голосов
/ 28 марта 2020

Я хочу, чтобы метод возвратил обратно pandas DataFrame, указав в качестве параметра строковый путь, который задается параметром конструктора.

Вот мой код:

class TestClass:
    def __init__(self,file):
        self.file = file
        print(self.loadfile(self.file))
        print("------------------------------------")
        print("DESCRIPTION: ", self.loadfile(self.file))


    def loadfile(self,file):
        df = pd.read_csv(self.file)
        return df

        return pd.DataFrame(columns=[file])

    def descript(self,file):

        df = pd.read_csv(self.file)
        return df.descript

Это возвращает мне что-то, но это не похоже на pandas фрейм данных, который вы получите, если наберете

df1 = pd.read_csv ("blabla.csv")

Так где ошибка?

Заранее спасибо!

1 Ответ

0 голосов
/ 28 марта 2020

Вы не можете вернуть DataFrame как часть __init__, вам нужно будет вызвать функцию loadfile после инициализации объекта класса

class TestClass(object):

    def __init__(self, file):
        self.file = file

    def loadfile(self):
        df = pd.read_csv(self.file)
        return df

TestClass(r'path\to\file.csv').loadfile()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...