Унаследованные методы класса возвращают родительский объект. Нужно заставить его возвращать объект дочернего класса ⇐ Python

[Anonymous]

Я пытаюсь проверить возможность создания подкласса класса PySpark DataFrame, чтобы добавить в свою работу дополнительные методы многократного использования.

Код: Выделить всё

class EnhancedDataframe(DataFrame):
def __init__(self, df):
super().__init__(df._jdf,df.sql_ctx)

def notNullCount(self,col_name):
return self.filter(col(col_name).isNotNull()).count()

Проблема в том, что если я использую это для создания новых фреймов данных, всякий раз, когда я использую родительский метод, который возвращает объект родительского типа, мои новые методы класса больше не будут доступны с ним.
Проблема в том, что если я использую это для создания новых фреймов данных, всякий раз, когда я использую родительский метод, который возвращает объект родительского типа, мои новые методы класса больше не будут доступны с этим.
p>
Итак, как вернуть объект дочернего класса для всех родительских методов. Один из вариантов, который я рассматриваю, — это использовать функцию __getattribute__ и попытаться вернуть дочерний объект, заключив его в имя дочернего класса. Есть ли у нас другие варианты? То, что я пытаюсь сделать, осуществимо ли это?

Подробнее здесь: https://stackoverflow.com/questions/743 ... bject-of-c