Pandas 来自 Pickle 的对象 - IDE 未知的类型

Pandas object from Pickle - Type not known by the IDE

在 Python 中,使用 IDE PyCharm,当我键入:

df = some_function_returning_a_dataframe()

然后 IDE 识别出 df 是一个 Pandas Dataframe 并在我稍后输入代码时给我有用的建议。

但是,如果我从 Pickle 导入数据框,通过:

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

然后 IDE 无法识别 df 的类型并且不再帮助我。 我假设 IDE 在阅读 pickle 之前不知道类型。

有什么办法可以解决吗?

你可以使用

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

帮助PyCharm知道df是一个DataFrame。

您也可以 assert 行中的类型,或者将 unpickling 放在带有提示 return 类型 (:rtype: pd.Dataframe) 的函数中,并且 PyCharm 将识别它