您的位置:首页 > 编程语言 > Python开发

解决python画图中colorbar设置刻度和标签字体大小

2019-04-22 14:46 1401 查看
  1. 介绍
    python很火,因为有各种库的支持,所以功能格外强大。在可视化方面,目前用得较多的是matplotlib.
    在基于matplotlib.pyplot画带色标(colorbar)的图时候,往往为了美观和科研用途,需要对colorbar的Ticks(刻度) ,标签(label)和fonddict(字体进行设置)。但是很多初学者都苦于这些东西的设置,因为太麻烦了(别问我怎么知道的)。以下将介绍有用的方法来解决这些问题。
import matplotlib.pyplot as plt
  1. 解决问题

下面将分两种情况来进行介绍

情况1:当colorbar在需要额外指定时(即不能自动生成)。
这种情况最为常见。如 plt.contourf ,plt.imshow, 等。即就是先基于plt.contourf生成图后,再获取该图的句柄,再设置colorbar。画主图和colorbar是两个语句。
还是举个例子吧

  • List item
import numpy as np
import matplotlib.pyplot as plt#设置一个随机种子,
#生成固定数值的随机整数数组
seed=np.random.seed(42)
data=np.random.randint(0,10,size=(10,10))

fig=plt.figure(figsize=(10,8))
#这就是所谓的第一种情况哦
h=plt.contourf(data)
cb=plt.colorbar(h)

生成的图如下所示

  • List item

我们设置一下主图的横纵坐标的刻度字体大小

plt.xticks(fontsize=16)
plt.yticks(fontsize=16)


看,在这种情况下,色标的刻度大小与主图的刻度大小不符合,这是完全不符合一个合格的程序员的审美的。因此需要更改colorbar刻度字体大小。

  • List item
fig=plt.figure(figsize=(10,8))
#这就是所谓的第一种情况哦
h=plt.contourf(data)
cb=plt.colorbar(h)
cb.ax.tick_params(labelsize=16)  #设置色标刻度字体大小。
plt.xticks(fontsize=16)
plt.yticks(fontsize=16)font = {'family' : 'serif',
'color'  : 'darkred',
'weight' : 'normal',
'size'   : 16,
}
cb.set_label('colorbar',fontdict=font) #设置colorbar的标签字体及其大小


这个图是不是漂亮多啦

  • List item

当你想改变colorbar刻度数值大小时,可以采用cb.set_ticks, cb.set_ticklabels
当然你也可以获取其刻度大小,可以用cb.get_ticks。还有cb.get_cmap可以获取颜色属性。

  • List item
    当你需要进行其他操作,如设置颜色,大小限制什么的,可以自己随意上手。

综上,最重要的一句是 cb.ax.tick_params(labelsize=16)。其他的属性也可以 用 cb.ax.+table键进行查看。

情况2:当colorbar可以在生成主图时就自动生成时
这种情况在生成热图时较为常见。
可以用seaborn来生成热图(heatmap)

import seaborn as sns
import matplotlib.pyplot as pltimport numpy as np
seed=np.random.seed(42)
data=np.random.randint(0,10,size=(10,10))

fig=plt.figure(figsize=(10,8))
h=sns.heatmap(data, annot=True,fmt='d',linewidths=0.5)

生成图如下:

heatmap的各个参数介绍可见如下链接:
http://seaborn.pydata.org/generated/seaborn.heatmap.html
https://www.geek-share.com/detail/2722060900.html
https://blog.csdn.net/m0_38103546/article/details/79935671
heatmap参数中,默认cbar=True,即默认画出colorbar,其中cbarkws为一个字典, 可以用来设置colorbar的一些属性,包括 thrink, orentation等 。但是我没有找到cbrkws可以设置哪些属性以及怎么设置。
因此这里不利用cbrkws进行colorbar设置。

fig=plt.figure(figsize=(10,8))
h=sns.heatmap(data, annot=True,fmt='d',linewidths=0.5,cbar=False) #设置不使用其默认自带的colorbar
cb=h.figure.colorbar(h.collections[0]) #显示colorbar
cb.ax.tick_params(labelsize=16) #设置colorbar刻度字体大小。
plt.xticks(fontsize=16)
plt.yticks(fontsize=16)


其他如标签和刻度范围和色标的操作同第一种情况。

参考链接:
https://www.geek-share.com/detail/2707628950.html
https://stackoverflow.com/questions/37233108/seaborn-change-font-size-of-the-colorbar
https://stackoverflow.com/questions/34820239/seaborn-heatmap-colorbar-label-as-percentage

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: