Pickle中的Pandas對象-IDE未知的類型

Question

在Python中，當我鍵入以下內容時，請使用IDE PyCharm：

df = some_function_returning_a_dataframe()

然后，IDE識別df是Pandas Dataframe，並在以后鍵入代碼時給我一些有用的建議。

但是，如果我從Pickle中導入數據框，則通過：

df = pandas.read_pickle("dump.pkl")

然后，IDE無法識別df的類型，並且不再對我有所幫助。 我假設IDE在讀取泡菜之前不知道類型。

有什么辦法可以應對嗎？

Answer 1

您也可以在行中assert類型，或者將:rtype: pd.Dataframe具有提示的返回類型（ :rtype: pd.Dataframe ）的函數中，PyCharm會識別出它

Answer 2

你可以用

df = pandas.DataFrame(pandas.read_pickle("dump.pkl"))

幫助PyCharm知道df是一個DataFrame。