matplotlib:一个图形上有多个图
问题内容:
我有一些代码:
import matplotlib.pyplot as plt
def print_fractures(fractures):
xpairs = []
ypairs = []
plt.figure(2)
plt.subplot(212)
for i in range(len(fractures)):
xends = [fractures[i][1][0], fractures[i][2][0]]
yends = [fractures[i][1][1], fractures[i][2][1]]
xpairs.append(xends)
ypairs.append(yends)
for xends,yends in zip(xpairs,ypairs):
plt.plot(xends, yends, 'b-', alpha=0.4)
plt.show()
def histogram(spacings):
plt.figure(1)
plt.subplot(211)
plt.hist(spacings, 100)
plt.xlabel('Spacing (m)', fontsize=15)
plt.ylabel('Frequency (count)', fontsize=15)
plt.show()
histogram(spacings)
print_fractures(fractures)
此代码将产生以下输出:
我的问题是:
1)为什么要创建两个单独的图形?我认为subplot命令会将它们组合成一个图形。我以为可能是多个plt.show()命令,但是我尝试将其注释掉,并且只从函数外部调用了一次,但仍然有2个窗口。
2)如何将它们正确组合成1个图形?另外,我希望图2的轴具有相同的比例(即,x轴上的400 m与y轴上的400 m长度相同)。同样,我也想垂直拉伸直方图-
这是如何完成的?
问题答案:
正如您已经观察到的,figure()
如果只打算使用一个图形(一个窗口),则无法在每个函数内部调用。相反,只需调用subplot()
而不show()
在函数内部调用即可。如果您处于模式,这show()
将强制pyplot
创建第二个图形plt.ioff()
。在plt.ion()
模式下,您可以将plt.show()
调用保留在本地上下文中(在函数内部)。
要在x和y轴上达到相同的比例,请使用plt.axis('equal')
。在下面,您可以看到此原型的说明:
from numpy.random import random
import matplotlib.pyplot as plt
def print_fractures():
plt.subplot(212)
plt.plot([1,2,3,4])
def histogram():
plt.subplot(211)
plt.hist(random(1000), 100)
plt.xlabel('Spacing (m)', fontsize=15)
plt.ylabel('Frequency (count)', fontsize=15)
histogram()
print_fractures()
plt.axis('equal')
plt.show()