Pythonic类型提示与pandas?

10 浏览
0 Comments

Pythonic类型提示与pandas?

让我们来看一个简单的函数,它接受一个字符串并返回一个DataFrame:

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

在这个函数中,添加类型提示的推荐Pythonic方式是什么?

如果我询问DataFrame的类型,Python会返回`pandas.core.frame.DataFrame`。

然而,下面的方式是行不通的,因为它会告诉我pandas未定义:

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

0