Колонка Pandas, записи которой принадлежат моему классу

Question

У меня есть фрейм данных A.

В нем есть столбец col.

A['col'] - это столбец, каждая запись которого имеет тип my_class.

'my_class' имеет атрибуты my_class.x и my_class.y

Я хочу разбить A['col'] на A['x'] и A['y'].

A['col'].str_split() не будет работать как тип записейне строка купить my_class

Как мне это сделать?

Answer 1

Вы можете использовать transform

df.col.transform(lambda k : pd.Series(k.__dict__))

---

Пример:

class A:
    def __init__(self, a, b):
        self.a = a
        self.b = b

obj1 = A("foo1", "bar1")
obj2 = A("foo2", "bar2")

df = pd.DataFrame({"col": [obj1, obj2]})

    col
0   <__main__.A object at 0x116de7b00>
1   <__main__.A object at 0x116de7438>

Тогда

df.col.transform(lambda k : pd.Series(k.__dict__))

    a       b
0   foo1    bar1
1   foo2    bar2