plt.cm.RdBu（在 cm.py 中找不到参考）[关闭]

Posted 2023-03-28

【中文标题】plt.cm.RdBu（在 cm.py 中找不到参考）[关闭]

【英文标题】：plt.cm.RdBu ( Cannot find reference in cm.py ) [closed]

【发布时间】：2021-11-01 13:25:41

【问题描述】：

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

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 中突出显示，并且警告消息显示：

在“cm.py”中找不到参考“RdBu”

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

【问题讨论】：

你检查过你的输入数据了吗？这看起来很像您的输入只是 1x1

@Eumel 输入数据没问题。所有方法都有效并表明输入正常。这是我关注的tutorial。我认为问题是plt.cm.RdBu 但不知道如何解决？

是否可以使用不同的颜色图？

你的意思是把heatmap换成不同的地图吗？

@CatarinaRuna 是的，完全正确

【参考方案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))