plt.cm.RdBu（无法在 cm.py 中找到参考）

Question

以下代码片段应该显示数据集的热图，如第一张图片：

colormap = plt.cm.RdBu
plt.figure(figsize=(18, 15))
plt.title('Pearson Correlation of Features', y=1.05, size=50)
sns.heatmap(df.corr(), linewidths=0.1, vmax=1.0, square=True, cmap=colormap, linecolor='white', annot=True)
plt.show()

但是，我的 RdBu 在 PyCharm 中突出显示并且警告消息显示：

Cannot find reference 'RdBu' in 'cm.py'

有几篇文章基本上表明语法是正确的，我没有遗漏任何包。我在这里做错了什么，我得到的是这张空地图？

Answer 1

我在图形代码之前缺少以下代码：

for c in df.columns:
    if df[c].dtype == 'object':
        lbl = LabelEncoder()
        lbl.fit(list(df[c].values))
        df[c] = lbl.transform(list(df[c].values))

plt.cm.RdBu（无法在 cm.py 中找到参考）

plt.cm.RdBu ( Cannot find reference in cm.py )

python

matplotlib

pycharm

heatmap

colormap