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

我的问题是:

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()