关于Pandas的Python式暗示?

2024-04-18 09:15:40 发布

您现在位置:Python中文网/ 问答频道 /正文

让我们使用一个简单的函数来获取str并返回一个数据帧:

import pandas as pd
def csv_to_df(path):
    return pd.read_csv(path, skiprows=1, sep='\t', comment='#')

向该函数添加类型提示的推荐pythonic方法是什么?

如果我向python请求数据帧的类型,它将返回pandas.core.frame.DataFrame。 不过,下面的方法行不通,因为它会告诉我熊猫是没有定义的。

 def csv_to_df(path: str) -> pandas.core.frame.DataFrame:
     return pd.read_csv(path, skiprows=1, sep='\t', comment='#')

Tags: csvto数据path函数pandasdfread
2条回答

为什么不直接使用pd.DataFrame

import pandas as pd
def csv_to_df(path: str) -> pd.DataFrame:
    return pd.read_csv(path, skiprows=1, sep='\t', comment='#')

结果相同:

> help(csv_to_df)
Help on function csv_to_df in module __main__:
csv_to_df(path:str) -> pandas.core.frame.DataFrame

我目前正在做以下工作:

from typing import TypeVar
PandasDataFrame = TypeVar('pandas.core.frame.DataFrame')
def csv_to_df(path: str) -> PandasDataFrame:
    return pd.read_csv(path, skiprows=1, sep='\t', comment='#')

它给出:

> help(csv_to_df)
Help on function csv_to_df in module __main__:

csv_to_df(path:str) -> ~pandas.core.frame.DataFrame

我不知道那有多像Python,但我发现,作为一个类型提示,这是可以理解的。

相关问题 更多 >