相关疑难解决方法(0)

具有混合数据类型的 pandas DataFrame 的类型提示

我一直在寻找 pandas DataFrame 的健壮类型提示,但似乎找不到任何有用的东西。这个问题仅仅触及了表面Pythonic 类型提示与 pandas?

通常,如果我想暗示以 DataFrame 作为输入参数的函数的类型,我会这样做:

import pandas as pd 
def func(arg: pd.DataFrame) -> int: 
     return 1
Run Code Online (Sandbox Code Playgroud)

我似乎找不到的是如何输入具有混合 dtypes 的DataFrame 提示。DataFrame 构造函数仅支持完整 DataFrame 的类型定义。因此,据我所知,数据类型的更改只能在该pd.DataFrame().astype(dtypes={})函数之后发生。

这在这里有效,但对我来说似乎不太Pythonic

import datetime
def func(arg: pd.DataFrame(columns=['integer', 'date']).astype(dtype={'integer': int, 'date': datetime.date})) -> int:
    return 1
Run Code Online (Sandbox Code Playgroud)

我遇到了这个包: https: //pypi.org/project/dataenforce/,其中包含如下示例:

def process_data(data: Dataset["id": int, "name": object, "latitude": float, "longitude": float])
  pass
Run Code Online (Sandbox Code Playgroud)

这看起来很有希望,但遗憾的是该项目已经过时且存在缺陷。

作为一名数据科学家,在构建具有长 ETL 流程的机器学习应用程序时,我认为类型提示非常重要。

你用什么,有人在 pandas 中输入暗示他们的数据框吗?

python type-hinting pandas

10
推荐指数
1
解决办法
9600
查看次数

标签 统计

pandas ×1

python ×1

type-hinting ×1