定位彩条

时间:2012-11-09 15:09:59

标签: python matplotlib

我有一个带有彩条的matplotlib图。我想将颜色条定位为水平,并在我的情节下面。

我几乎通过以下方式完成了这项工作:

plt.colorbar(orientation="horizontal",fraction=0.07,anchor=(1.0,0.0))

但是颜色条仍然略微重叠(和x轴的标签)。我想进一步向下移动颜色条,但我无法弄清楚如何做到这一点。

2 个答案:

答案 0 :(得分:53)

获得对色条位置的良好控制的最佳方法是给它自己的轴。像这样:

# What I imagine your plotting looks like so far
fig = plt.figure()
ax1 = fig.add_subplot(111)
ax1.plot(your_data)

# Now adding the colorbar
cbaxes = fig.add_axes([0.8, 0.1, 0.03, 0.8]) 
cb = plt.colorbar(ax1, cax = cbaxes)  

add_axes方括号中的数字指的是[left,bottom,width,height],其中坐标只是从绘图区域的0到1的分数。

答案 1 :(得分:48)

使用填充pad

为了相对于子图移动颜色条,可以使用pad参数fig.colorbar

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(1)

fig, ax = plt.subplots(figsize=(4,4))
im = ax.imshow(np.random.rand(11,16))
ax.set_xlabel("x label")

fig.colorbar(im, orientation="horizontal", pad=0.2)
plt.show()

enter image description here

使用轴分隔符

可以使用make_axes_locatable的实例来划分轴并创建一个与图像绘图完美对齐的新轴。同样,pad参数将允许设置两个轴之间的空间。

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable
import numpy as np; np.random.seed(1)

fig, ax = plt.subplots(figsize=(4,4))
im = ax.imshow(np.random.rand(11,16))
ax.set_xlabel("x label")

divider = make_axes_locatable(ax)
cax = divider.new_vertical(size="5%", pad=0.7, pack_start=True)
fig.add_axes(cax)
fig.colorbar(im, cax=cax, orientation="horizontal")

plt.show()

enter image description here

使用子图

可以直接创建两行子图,一行用于图像,另一行用于颜色条。然后,在图形创建中将height_ratios设置为gridspec_kw={"height_ratios":[1, 0.05]},使得其中一个子图的高度远小于另一个子图,这个小的子图可以托管颜色条。

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(1)

fig, (ax, cax) = plt.subplots(nrows=2,figsize=(4,4), 
                  gridspec_kw={"height_ratios":[1, 0.05]})
im = ax.imshow(np.random.rand(11,16))
ax.set_xlabel("x label")

fig.colorbar(im, cax=cax, orientation="horizontal")

plt.show()

enter image description here