可视化技巧：显示带colorbar的热图（matplotlib）

此人姓于名叫罩百灵

已于 2022-03-18 17:31:18 修改

阅读量2.5k

点赞数 2

分类专栏：机器学习文章标签：可视化 matplotlib 热图 colorbar

于 2020-04-13 17:06:37 首次发布

本文链接：https://blog.csdn.net/xuyangcao123/article/details/105492425

版权

机器学习专栏收录该内容

25 篇文章 4 订阅

订阅专栏

需求

在热图边画出colorbar，并设置一些参数使得colorbar美观。

然后保存图像。

代码:

def save_jet_figure(image, filename):
    ''' save jet colormap figure using matplotlib
    
    Args: 
    ---

        image: ndarray of (width, height)
            input image in gray scale.

        filename: str
            save filename

    Returns:
    ---

        None

    Example:
    --- 

        from skimage.io import imread
        import matplotlib.pyplot as plt
        from mpl_toolkits.axes_grid1 import make_axes_locatable

        img = imread('./test.png')
        img = img[:, :, 0]
        save_jet_figure(img, './result.png') 

        plt.show()
        
    '''
    # show image
    fig = plt.figure(tight_layout=True)
    ax = fig.add_subplot(111)
    im = ax.imshow(image, 'jet')
    ax.axis('off')

    # colorbar
    divider = make_axes_locatable(ax)
    cax = divider.append_axes('right', size='2%', pad=0.04)
    cbar = plt.colorbar(im, 
                        cax=cax, 
                        extend='both', 
                        extendrect=True, 
                        ticks=[0.0, 0.5, 1.0],
                        )
    cbar.outline.set_visible(False)
    cbar.ax.tick_params(labelsize=8, 
                        width=0,
                        length=0,
                        pad=1,)

    # save image
    fig.savefig(filename, 
                bbox_inches='tight',
                pad_inches=0,
                transparent=True,
                dpi=300)