Есть ли способ создать пустой фрейм данных 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
Создать пустой фрейм данных pandas из pandera DataFrameModel ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение
-
-
Может ли Pandera Python проверить, имеет ли фрейм данных Geopandas строки строк?
Anonymous » » в форуме Python - 0 Ответы
- 26 Просмотры
-
Последнее сообщение Anonymous
-