Pycharm 数据帧连接后不输入提示

Question

如果我想将 2 个数据帧连接成 1 个：

X_all = pd.concat(objs = [X_train, X_test], axis=0) # type: pd.DataFrame
X_all.fillna(X_all.mean(), inplace = True)

即使我在评论中告诉 pycharm 类型 pd.Dataframe，在第二行它仍然不会输入提示 .fillna 并且我仍然收到警告：Unresolved attribute reference 'fillna' for class 'type' less... (⌘F1)

如何让 pycharm 识别它是一个数据帧？

Answer 1

我必须做的：

X_all = pd.concat(objs = [X_train, X_test], axis=0)
assert isinstance(X_all, pd.DataFrame)
X_all.fillna(X_all.mean(), inplace = True)

Answer 2

你的类型提示是错误的。应该是pd.DataFrame，不是pd.Dataframe（大写F）。

X_all = pd.concat(objs = [X_train, X_test], axis=0)  # type: pd.DataFrame

在 Python 3.6+ 中，您还可以使用本机类型提示：

X_all: pd.DataFrame = pd.concat(objs = [X_train, X_test], axis=0)

Pycharm does not type hint after dataframe concat