Pandas: Поддержка класса данных

Созданный на 14 июл. 2018 · 3Комментарии · Источник: pandas-dev/pandas

Описание предложения

Классы данных были добавлены в Python 3.7.

Было бы неплохо, если бы панды поддерживали классы данных. Например, можно построить кадр данных, вызвав .from_dataclasses или просто .DataFrame(data=dataclass_list) . Также должна быть возможность делать .to_dataclasses .

Ожидаемое поведение

from dataclasses import dataclass
import pandas as pd

<strong i="12">@dataclass</strong>
class SimpleDataObject(object):
  field_a: int
  field_b: str

dataclass_object1 = SimpleDataObject(1, 'a')
dataclass_object2 = SimpleDataObject(2, 'b')
>>> asd

# Dataclasses to DataFrame
df = pd.from_dataclasses([dataclass_object1, dataclass_object2])
df.dtypes == ['field_a', 'field_b']
>>> True
df.dtypes == ['int', 'str']
>>> True

# Dataclasses to DataFrame
df = pd.DataFrame(data=[dataclass_object1, dataclass_object2])
df.dtypes == ['field_a', 'field_b']
>>> True
df.dtypes == ['int', 'str']
>>> True

# DataFrame to Dataclasses
df = pd.DataFrame(columns=['field_a', 'field_b'], data=[[1, 'a'], [2, 'b']])
dataclass_list = df.to_dataclasses()
dataclass_list == [dataclass_object1, dataclass_object2]
>>> True

API Design Needs Discussion

Источник

aidiss

👍25

Самый полезный комментарий

Модуль dataclasses имеет функции самоанализа is_dataclass и fields , так что эта часть не должна быть проблемой.

Тем не менее, я не уверен, что нам следует быстро переходить к какому-либо конкретному API/поддержке здесь. На данный момент хелпер asdict из модуля dataclasses может помочь со сценарием загрузки.

In [18]: from dataclasses import asdict

In [19]: pd.DataFrame([asdict(x) for x in [dataclass_object1, dataclass_object2]])
Out[19]:
   field_a field_b
0        1       a
1        2       b

chris-b1 15 июл. 2018

👍34 🎉6

Все 3 Комментарий

AFAIK не гарантирует, что вы можете знать, что определенный экземпляр является классом данных. Например, классы не наследуются от класса данных.

Из вашего примера:

<strong i="7">@dataclass</strong>
class SimpleDataObject(object):
  field_a: int
  field_b: str

x = SimpleDataObject(a=2, b=‘f’)

Я не думаю, что вы могли бы даже сказать из самоанализа, что x — это класс данных, верно? Если это так, то это невозможно сделать.

topper-123 15 июл. 2018

Модуль dataclasses имеет функции самоанализа is_dataclass и fields , так что эта часть не должна быть проблемой.

In [18]: from dataclasses import asdict

In [19]: pd.DataFrame([asdict(x) for x in [dataclass_object1, dataclass_object2]])
Out[19]:
   field_a field_b
0        1       a
1        2       b

chris-b1 15 июл. 2018

👍34 🎉6

Я скомпилировал решение, в котором я проверяю данные, предоставленные во время __init__, в этом PR , однако, похоже, что их конвейер тестирования настроен для поддержки нескольких py-версий. Так что мне может понадобиться немного больше времени, чтобы это произошло.