如何更改颜色条以与 Matplotlib 中的主图对齐？

Question

在 Matplotlib 中使用 imshow 绘制矩阵时，colorbar 图例条的大小、位置、字体等参数如何更改？

这里我创建了一个示例代码

import numpy as np
import matplotlib
import matplotlib.pyplot as plt
%matplotlib inline

def plot_matrix(mat, title='example', cmap=plt.cm.Blues):
    plt.imshow(mat, interpolation='nearest', cmap=cmap)
    plt.grid(False)
    plt.title(title)
    plt.colorbar()

data = np.random.random((20, 20))

plt.figure(figsize=(8,8))
plt.tick_params(axis='both', which='major', labelsize=12)

plot_matrix(data) 

在一个真实的用例中，我得到了复杂的标签，图例栏变得比矩阵本身高得多。我想更改图例栏以使情节更有效地利用空间。

我为matplotlib.pyplot.colorbar 找到了documentation，但是还没有找到设置颜色图例栏的大小、位置和字体大小的好方法。

Answer 1

imshow 强制执行 1:1 方面（默认情况下，但您可以使用 aspect 参数更改它），这使事情变得有点棘手。为了始终获得一致的结果，我可能会建议手动指定轴的大小：

import numpy as np
import matplotlib
import matplotlib.pyplot as plt
%matplotlib inline

def plot_matrix(mat, figsize, title='example', cmap=plt.cm.Blues):
    f = plt.figure(figsize=figsize)
    ax = plt.axes([0, 0.05, 0.9, 0.9 ]) #left, bottom, width, height
    #note that we are forcing width:height=1:1 here, 
    #as 0.9*8 : 0.9*8 = 1:1, the figure size is (8,8)
    #if the figure size changes, the width:height ratio here also need to be changed
    im = ax.imshow(mat, interpolation='nearest', cmap=cmap)
    ax.grid(False)
    ax.set_title(title)
    cax = plt.axes([0.95, 0.05, 0.05,0.9 ])
    plt.colorbar(mappable=im, cax=cax)
    return ax, cax

data = np.random.random((20, 20))
ax, cax = plot_matrix(data, (8,8)) 

现在您有了绘制颜色条的轴cax。你可以做很多事情，比如旋转标签，使用plt.setp(cax.get_yticklabels(), rotation=45)