如何将Pandera的pa.DataFrameModel转为pa.SeriesSchema以验证DataFrame行？

核心实现思路 #

直接复用DataFrameModel的__fields__属性（包含所有字段的验证规则），动态构建SeriesSchema，无需手动重复定义字段结构。

完整代码示例 #

1. 定义DataFrameModel #

import pandera as pa
from pandera.typing import DataFrame, Series
import pandas as pd

class MyDataframeSchema(pa.DataFrameModel):
    id: Series[int] = pa.Field(ge=1)
    name: Series[str] = pa.Field(min_length=2)
    score: Series[float] = pa.Field(ge=0, le=100)

2. 动态生成SeriesSchema #

# 从DataFrameModel提取字段规则，构建基础SeriesSchema
series_schema = pa.SeriesSchema(
    {field_name: field.schema for field_name, field in MyDataframeSchema.__fields__.items()}
)

# 若需要包含索引验证，可添加index参数同步DataFrameModel的索引规则
series_schema_with_index = pa.SeriesSchema(
    {field_name: field.schema for field_name, field in MyDataframeSchema.__fields__.items()},
    index=MyDataframeSchema.__index_schema__
)

3. 验证单行数据 #

# 创建示例DataFrame
df = pd.DataFrame({
    "id": [1, 2],
    "name": ["Alice", "Bob"],
    "score": [85.5, 92.0]
})

# 提取单行并执行验证
single_row = df.iloc[0]
validated_row = series_schema.validate(single_row)
print(validated_row)

优势说明 #

• 完全复用已有DataFrameModel的验证规则，遵循**DRY（Don't Repeat Yourself）**原则
• 自动同步DataFrameModel的规则变更，无需手动维护两套独立的schema
• 支持包含索引验证的场景，覆盖完整的行数据校验需求

内容的提问来源于stack exchange，提问作者codekoriko