我使用gridspec绘制两个数字:

fig = plt.figure(num=2,figsize=(5,2))

gs = gridspec.GridSpec(1, 2, width_ratios=[1,1])

ax0 = plt.subplot(gs[0])

ax1 = plt.subplot(gs[1])

ax0.imshow(ROI,'gray')

map1 = ax1.imshow(z,cmap=cm.YlOrRd)

divider1 = make_axes_locatable(ax1)

cax1 = divider1.append_axes("right", size="5%", pad=0.05)

cbar1 = plt.colorbar(map1, cax=cax1)

ROI和z预先定义了相同的大小.

但他们的尺寸略有不同.

我想把它做成相同的尺寸:

Pniie.png

怎么解决这个问题?

解决方法:

问题是您在gridspec中将两个图形的宽度固定为相同,然后将颜色条添加到第二个图形中,该图形占用了一些宽度(5%加上填充).并且由于imshow具有1:1的宽高比,这种宽度变化也会改变图形的高度.

避免这种情况的最简单方法是在宽度比中考虑颜色条的额外宽度:

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.gridspec as gridspec

from mpl_toolkits.axes_grid1 import make_axes_locatable

# Random test data:

ROI = np.random.rand(250, 150)

z = np.random.rand(250, 150)

fig = plt.figure(num=2,figsize=(4,2))

gs = gridspec.GridSpec(1, 2, width_ratios=[1,1.08])

ax0 = plt.subplot(gs[0])

ax1 = plt.subplot(gs[1])

ax0.imshow(ROI,'gray')

map1 = ax1.imshow(z,cmap=plt.cm.YlOrRd)

divider1 = make_axes_locatable(ax1)

cax1 = divider1.append_axes("right", size="5%", pad="3%")

cbar1 = plt.colorbar(map1, cax=cax1)

plt.show()

这将导致以下情节:

mNpWd.png

您还可以通过使用gridspec为颜色栏添加第三个轴来避免这种情况:

import numpy as np

import matplotlib.pyplot as plt

import matplotlib.gridspec as gridspec

# Random test data:

ROI = np.random.rand(250, 150)

z = np.random.rand(250, 150)

fig = plt.figure(num=2,figsize=(5,2))

gs = gridspec.GridSpec(1, 3, width_ratios=[1,1,0.05])

ax0 = plt.subplot(gs[0])

ax1 = plt.subplot(gs[1])

cax1 = plt.subplot(gs[2])

ax0.imshow(ROI,'gray')

map1 = ax1.imshow(z,cmap=plt.cm.YlOrRd)

cbar1 = plt.colorbar(map1, cax=cax1)

plt.show()

这给出了:

Bkpmn.png

这可能更灵活,但也需要更多的填充和尺寸调整.

标签:python,matplotlib

来源: https://codeday.me/bug/20190929/1833069.html

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐