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。