Seaborn 对角相关矩阵跳过第一行和最后一列

Question

如何从下图中删除标签和相应的列/行（空白）？在这种特殊情况下，in 将对应于行 a 和列 l。

mask = np.zeros_like(corr, dtype=np.bool)
mask[np.triu_indices_from(mask)] = True

f, ax = plt.subplots(figsize=(12, 10))
cmap = sns.diverging_palette(220, 10, as_cmap=True)

corr = scaled_df.select_dtypes(include = ['float64', 'int64']).corr()

sns.heatmap(corr, mask=mask, cmap='inferno', vmin = -0.1, vmax=0.8, square=True)
plt.axvline(1.0, color='white', linestyle = '-', linewidth = 2.5, alpha = 1.0)

Answer 1

这里有两种相当简单的方法来解决这个问题：

A.将有效/期望的数据子集传递给绘图仪

corr = np.random.rand(10,10)
mask = np.zeros_like(corr, dtype=np.bool)
mask[np.triu_indices_from(mask)] = True

sns.heatmap(corr[1:,:-1], mask=mask[1:,:-1], cmap='inferno', vmin = -0.1, vmax=0.8, square=True)

B.渲染后，调整绘图范围

ax = sns.heatmap(corr, mask=mask, cmap='inferno', vmin = -0.1, vmax=0.8, square=True)
ax.set_xlim(0,9)
ax.set_ylim(10,1)

尚不清楚您是否总是有三角形有效区域，但您可以检测到 根据掩码中设置的所有行，掩码中的矩阵，等效于 列。但对于源自 triu（或 tril）的有效区域，限制仅为开始/结束 1 处。