matplotlib:一个图上的多个图

时间:2014-01-23 23:51:29

标签: python matplotlib plot

我有一些代码:

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)

此代码将产生以下输出: Fig1

我的问题是:

1)为什么要创建两个独立的数字?我认为subplot命令会将它们组合成一个图形。我认为它可能是多个plt.show()命令,但我尝试将这些命令注释掉,只从我的函数外部调用一次,我仍然有2个窗口。

2)如何将它们正确地组合成1个图?另外,我希望图2的轴具有相同的比例(即,x轴上的400米与y轴上的400米的长度相同)。同样,我也想垂直拉伸直方图 - 这是如何实现的?

1 个答案:

答案 0 :(得分:8)

正如您所观察到的,如果您打算只使用一个数字(一个窗口),则无法在每个函数内调用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()