我尝试得到以下内容,如下图所示。
让我们假设,为简单起见,我有一个numpy数组(10x10),我想用matplotlib imshow绘制。条件是具有不同的像素大小,例如:前五行的大小应为0.5厘米,后五行的大小应为1厘米。列应具有相同的大小。
我怎么能轻松实现这个?我已经尝试过这样做,但我不喜欢这个解决方案;特别是我仍然有白色边框,变焦很可怕。
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()
答案 0 :(得分: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
个对象