来自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在读取pickle之前不知道类型。

有没有办法解决这个问题？

Answer 1

You can also assert the type in line, or put the unpickling in a function with a hinted return type (:rtype: pd.Dataframe), and PyCharm will recognize it

Answer 2

您可以使用

/log

帮助PyCharm知道df = pandas.DataFrame(pandas.read_pickle("dump.pkl")) 是一个DataFrame。