Создать пустой фрейм данных pandas из pandera DataFrameModelPython

Программы на Python
Ответить
Anonymous
 Создать пустой фрейм данных pandas из pandera DataFrameModel

Сообщение Anonymous »

Есть ли способ создать пустой фрейм данных pandas из схемы pandera?
Учитывая следующую схему, я бы хотелось бы получить пустой фрейм данных, как показано ниже:
from pandera.typing import Series, DataFrame

class MySchema(pa.DataFrameModel):
state: Series[str]
city: Series[str]
price: Series[int]

def get_empty_df_of_schema(schema: pa.DataFrameModel) -> pd.DataFrame:
pass

wanted_result = pd.DataFrame(
columns=['state', 'city', 'price']
).astype({'state': str, 'city': str, 'price': int})
wanted_result.info()

Желаемый результат:
Index: 0 entries
Data columns (total 3 columns):
# Column Non-Null Count Dtype
--- ------ -------------- -----
0 state 0 non-null object
1 city 0 non-null object
2 price 0 non-null int64


Изменить:
Найдено рабочее решение:
def get_empty_df_of_pandera_model(model: [DataFrameModel, MetaModel]) -> pd.DataFrame:
schema = model.to_schema()
column_names = list(schema.columns.keys())
data_types = {column_name: column_type.dtype.type.name for column_name, column_type in schema.columns.items()}
return pd.DataFrame(columns=column_names).astype(data_types)


Подробнее здесь: https://stackoverflow.com/questions/766 ... framemodel
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

Вернуться в «Python»