Добавьте свойство списка в подкласс pd.DataFrame. ⇐ Python

[Anonymous]

Я использую подкласс класса DataFrame pandas. Подкласс должен иметь свойство, представляющее собой список. Вот пример:

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

import pandas as pd

class MyDataFrame(pd.DataFrame):
def __init__(self, data, colors, *args, **kwargs):
m = pd.DataFrame(data)
super().__init__(m, *args, **kwargs)
self.colors = colors

my_df = MyDataFrame(
{
"name": ["Fred", "Wilma"],
"age": [42, 38]
},
colors=["red", "yellow", "green"])

При этом появляется следующее предупреждение о self.colors =colors:

Предупреждение пользователя: Pandas не поддерживает 'не разрешать создание столбцов с помощью нового имени атрибута - см. https://pandas.pydata.org/pandas-docs/s ... ute-access
Похоже, что проблема в том, что DataFrame обрабатывает заголовки столбцов как атрибуты и интерпретирует "

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

self.colors = colors

" в качестве запроса на добавление столбца в DataFrame, от чего вполне разумно отказаться. Я пытался добавить установщик, но безрезультатно. Я также пытался переместить назначение атрибута над super().__init__< /code>, но оказался в бесконечной рекурсии. Что я могу сделать, чтобы это исправить?

Подробнее здесь: https://stackoverflow.com/questions/792 ... e-subclass