python - Imshow - 用不同的像素大小分割

Question

我尝试得到以下内容，应如下图所示。

为了简单起见，让我们假设我有一个 numpy 数组(10x10)，我想用 matplotlib imshow 绘制它。条件是具有不同的像素大小，例如:前五行的大小应为0.5cm，后五行的大小应为1cm。列应具有相同的大小。

我怎样才能轻松实现这个？我已经尝试过这样做，但我不喜欢这个解决方案；特别是我仍然有白色边框，缩放效果很糟糕。

from matplotlib import pyplot as pl
import numpy as np

data = np.arange((100))
data = np.reshape(data, (10,10))

figure, (ax1, ax2) = pl.subplots(2, 1, sharex='col')
figure.subplots_adjust(hspace=0)
data1=data[5:10,:]
ax1.imshow(data1, origin="lower", interpolation="none", aspect=0.5, extent=[-0.5,10.5,5.5,10.5], vmax=np.amax(data), vmin=np.amin(data))
ax1.set_ylim([5.5,10.5])
##
data2=data[0:5,:]
ax2.imshow(data2, origin="lower", interpolation="none", aspect=1, extent=[-0.5,10.5,-0.5,5.5], vmax=np.amax(data), vmin=np.amin(data))
ax2.set_ylim([-0.5,5.5])
pl.show()

谢谢

Answer 1

如果您只使用单个轴对象，这会更简单。然后缩放也可以完美地工作。

代码:

from matplotlib import pyplot as plt
import numpy as np

# prepare the data
data = np.arange((100))
data = np.reshape(data, (10,10))
data1=data[0:5,:]
data2=data[5:10,:]

# create the figure and a single axis
fig, ax = plt.subplots()

# common arguments to imshow
kwargs = dict(
        origin='lower', interpolation='nearest', vmin=np.amin(data),
        vmax=np.amax(data), aspect='auto')

# draw the data
ax.imshow(data1, extent=[0, 10, 0, 5], **kwargs)
ax.imshow(data2, extent=[0, 10, 5, 7.5], **kwargs)

# optional black line between data1 and data2
ax.axhline(5, color='k')

# set the axis limits
ax.set_ylim(0, 7.5)
ax.set_xlim(0, 10)

# set the xticklabels
xticks = np.arange(0,10)
ax.set_xticks(xticks + 0.5)
ax.set_xticklabels(map(str, xticks))

# set the yticks and labels
yticks = np.concatenate((
        np.arange(0, 5) + 0.5,
        np.arange(5, 7.5, 0.5) + 0.25
        ))
ax.set_yticks(yticks)
ax.set_yticklabels(map(str, xticks))

# show the figure
plt.show()

结果:

评论:

• 我冒昧地以更直观的方式重命名了 data1/2 对象
• 感谢 @kazemakase 指出需要调整轴刻度。