Matplotlib在imshow上绘制,同时保持轴大小

时间:2017-11-30 19:41:16

标签: matplotlib plot overlay axis imshow

每当我使用imshow()绘制图像时,在双轴底部x轴上绘制1D数据会改变用imshow()创建的初始x轴的大小和纵横比。我该如何避免这种行为?以下是重现此问题的方法:

import numpy as np
import matplotlib
matplotlib.use('macosx')
import matplotlib.pyplot as plt

im = np.random.rand(2856, 4290)
light_curve = im[1000, :]

fig = plt.figure(1, figsize=(10,10))
ax1 = plt.subplot(2,1,1)
ax1.imshow(im, cmap='gray', origin='lower')
ax2 = plt.subplot(2,1,2)
ax2.imshow(im, cmap='gray', origin='lower')
# Setting aspect ratio to equal does not help
ax2.set_aspect('equal')

ax21 = ax2.twinx()
ax21.plot(light_curve, alpha=0.7)
# Setting axis limits does not help
ax1.axis([0, im.shape[1], 0, im.shape[0]])
ax21.set_xlim([0, im.shape[1]])

以下是我的图形后端(macosx,如果它有任何相关性)的样子。

Top: image with imshow(). Bottom: plot overlaid, image axis aspect and size changed

首先使用twinx()用于帮助解决这个问题的目的不是吗? 那么我怎样才能保持初始imshow() x轴固定并使1D图的后续轴简单适合,而不需要调整宽松比例,而不需要完全手动构建我的轴?

1 个答案:

答案 0 :(得分:1)

确实有点不幸的是,这个方面不会传播到双轴,因为它周围会有相同的方框。

我认为解决这个问题的唯一方法是手动计算方面并将其设置为双轴。

import numpy as np
import matplotlib.pyplot as plt

im = np.random.rand(285, 429)
light_curve = im[100, :]

fig = plt.figure(1, figsize=(8,8))
ax1 = plt.subplot(2,1,1)
ax1.imshow(im, cmap='gray', origin='lower')
ax2 = plt.subplot(2,1,2)

ax2.imshow(im, cmap='gray', origin='lower')
ax2.set_aspect("equal", "box-forced")

ax21 = ax2.twinx()

ax21.plot(light_curve, alpha=0.7)
# Setting axis limits does not help
ax21.set_xlim(ax1.get_xlim())

a = np.diff(ax21.get_ylim())[0]/np.diff(ax1.get_xlim())*im.shape[1]/im.shape[0]
ax21.set_aspect(1./a, "box-forced")

plt.show()

enter image description here