pandas 如何在Pandera中使用SchemaModel验证XML索引

baubqpgj 于 2023-11-15 发布在其他

关注(0)|答案(3)|浏览(225)

我可以像这样使用DataFrameSchema验证DataFrame索引：

import pandera as pa
from pandera import Column, DataFrameSchema, Check, Index
schema = DataFrameSchema(
    columns={
        "column1": pa.Column(int),
    },
    index=pa.Index(int, name="index_name"),
)
# raises the error as expected
schema.validate(
    pd.DataFrame({"column1": [1, 2, 3]}, index=pd.Index([1, 2, 3], name="index_incorrect_name")) 
)

字符串
有没有一种方法可以使用SchemaModel来做同样的事情？

pandas

来源：https://stackoverflow.com/questions/73262749/how-to-validate-a-dataframe-index-using-schemamodel-in-pandera

3条答案

按热度按时间

qybjjes11#

找到an answer in GitHub
可以使用pa.typing.Index对索引进行类型注解。

class Schema(pa.SchemaModel):
    column1: pa.typing.Series[int]
    index_name: pa.typing.Index[int] = pa.Field(check_name=True)

字符串
了解如何验证MultiIndex索引：https://pandera.readthedocs.io/en/stable/schema_models.html#multiindex

赞(0）回复(0）举报 2023-11-15

wf82jlnq2#

你可以这样做-

import pandera as pa
from pandera.typing import Index, Series
class Schema(pa.SchemaModel):
    idx: Index[int] = pa.Field(ge=0, check_name=True)
    column1: Series[int]
df = pd.DataFrame({"column1": [1, 2, 3]}, index=pd.Index([1, 2, 3], name="index_incorrect_name")) 
Schema.validate(df)