[英]How to remove gaps between *images* in matplotlib?
受到这个问题的启发,我一直在尝试绘制没有间隙的图像。
在我的玩具示例中,我有四个图像要放在两行中。 它们有不同的形状:不同的行数,相同的列数。 尽管存在差异,但它们应该可以无间隙地放在一个图中,如下图所示:
但是,当我尝试将它们组合在一起时,设置plt.subplots_adjust(wspace=0, hspace=0)
不起作用,因为图像具有不同的形状。
这是代码:
from numpy.random import rand
import matplotlib.pyplot as plt
test_data = [[rand(10,10), rand(10,10)],[rand(5,10), rand(5,10)]]
f, axarr = plt.subplots(2,2)
for i in range(2):
for j in range(2):
axarr[i, j].imshow(test_data[i][j])
plt.tight_layout()
plt.subplots_adjust(wspace=0, hspace=0)
plt.show()
我试过玩set_aspect
和equal
,但没有运气。
有谁知道如何消除这些差距?
您可以设置height_ratios
使用次要情节的gridspec_kw
在调用参数plt.subplots
,并使用不同的图像的高度来设置比例。
从文档到plt.subplots()
:
gridspec_kw : 字典,可选
带有传递给 GridSpec 构造函数的关键字的字典,用于创建放置子图的网格。
请注意,您的示例具有默认 matplotlib 图像大小的确切纵横比,因此在添加更多行或更改图像形状之前,您不会注意到出现任何间隙。
因此,要将其扩展为通用解决方案,您需要根据图像的形状设置图形大小。 例如,让我们将示例扩展为 3 行 2 列。 我们还将明确地将图形宽度设置为 8 英寸,并根据图像大小调整高度。
from numpy.random import rand
import matplotlib.pyplot as plt
test_data = [[rand(10,10), rand(10,10)],[rand(5,10), rand(5,10)],[rand(2,10), rand(2,10)]]
cmaps = [['viridis', 'binary'], ['plasma', 'coolwarm'], ['Greens', 'copper']]
heights = [a[0].shape[0] for a in test_data]
widths = [a.shape[1] for a in test_data[0]]
fig_width = 8. # inches
fig_height = fig_width * sum(heights) / sum(widths)
f, axarr = plt.subplots(3,2, figsize=(fig_width, fig_height),
gridspec_kw={'height_ratios':heights})
for i in range(3):
for j in range(2):
axarr[i, j].imshow(test_data[i][j], cmap=cmaps[i][j])
axarr[i, j].axis('off')
plt.subplots_adjust(wspace=0, hspace=0, left=0, right=1, bottom=0, top=1)
plt.show()
您可以将所有图像组合成一个数组,然后使用imshow
绘制该数组。
为了确保图像周围没有多余的间距,我们需要计算纵横比并相应地设置图形大小。 (为了表明我将较低图像的像素数更改为 4)。
from numpy.random import rand
import matplotlib.pyplot as plt
import numpy as np
test_data = [[rand(10,10), rand(10,10)],[rand(4,10), rand(4,10)]]
a = np.c_[np.array(test_data[0][0]),np.array(test_data[0][1]) ]
b = np.c_[np.array(test_data[1][0]),np.array(test_data[1][1]) ]
c = np.r_[a,b]
dpi=100
width = 5 #inch
height = width*c.shape[0]/float(c.shape[1])
fig, ax = plt.subplots(figsize=(width,height ), dpi=dpi)
ax.imshow(c)
ax.axis("off")
plt.subplots_adjust(0,0,1,1)
plt.savefig(__file__+".png")
plt.show()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.