1. ホーム
  2. python

[解決済み] pandasでPythonicな型ヒントを?

2022-10-18 20:32:32

質問

strを受け取ってデータフレームを返す簡単な関数を考えてみましょう。

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

この関数に型ヒントを追加するための推奨される pythonic 方法は何ですか?

もし私がpythonにDataFrameの型を尋ねると、次のようなものが返ってきます。 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='#')

どのように解決するのですか?

なぜ 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