如何為 Python 中的類型提示導入 class？

Question

我有一個代碼：

from pandas import DataFrame
def func(df: DataFrame) -> DataFrame
    ...

因為我實際上並沒有在我的代碼中創建 DataFrame object 我想做：

from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from pandas import DataFrame
def func(df: DataFrame) -> DataFrame
    ...

這會引發錯誤：

E   NameError: name 'DataFrame' is not defined

我該如何解決？ 我必須保留原始進口嗎？ 導入 package 對我來說沒有什么意義，我的代碼並不是真正需要的，只是為了提示

Answer 1

您將希望使用from __future__ import annotations ，以便在運行時不嘗試評估表達式； 注釋將被視為隱式引用的字符串。