擅长:python、mysql、java
<p>如果我理解正确,您的意图是为<code>pd.Dataframe</code>创建pythonic类型提示。我想您可以利用以下实现:</p>
<pre><code>import pandas as pd
from pydantic import BaseModel
from typing import TypeVar
PandasDataFrame = TypeVar('pandas.core.frame.DataFrame')
class SubModelInput(BaseModel):
a: PandasDataFrame
b: PandasDataFrame
class ModelInput(BaseModel):
SubModelInput: SubModelInput
a: PandasDataFrame
b: PandasDataFrame
c: PandasDataFrame
data_frame = pd.DataFrame([{"a": "foo", "b": "bar"}])
sub_model = SubModelInput(a=data_frame, b=data_frame)
model = ModelInput(a=data_frame, b=data_frame, c=data_frame, SubModelInput=sub_model)
model.dict()
# {'SubModelInput': {'a': a b
# 0 foo bar, 'b': a b
# 0 foo bar}, 'a': a b
# 0 foo bar, 'b': a b
# 0 foo bar, 'c': a b
# 0 foo bar}
</code></pre>