在一个图上对另一个图绘制两个数组

Question

我想比较两个自变量随时间的变化，把它们都标在一个图上。这三个变量都是数组形式的，我是从一个文本文件中提取出来的。 这就是我目前所得到的：

from pylab import *

data_ = []

with open('all_the_data.txt') as dat_:
    for line in dat_:
        data_.append([i for i in line.split()])

D = zip(*data_)

def f1(t):
    y = D[1]
    return y

def f2(t):
    y = D[2]
    return y

if __name__ == '__main__':
    t = D[0]
    A = f1
    B = f2
    plot(t, A, 'bo')
    hold('on')
    plot(t, B, 'gX')
    xlabel('timestamp (unix)')
    ylabel('Station population')
    legend('Station 1','Station 2')
    title('Variance of Stations 1 and 2')
    show()
    savefig('2_stations_vs_time.png')

问题是，它不起作用，我不知道为什么。我是从一个绘制两个函数的教程中得到的。

Answer 1

我们绘制的是数据而不是函数。所以passA，B是错误的。我想你需要做的是：

from pylab import *

data_ = []

with open('all_the_data.txt') as dat_:
    for line in dat_:
        data_.append([i for i in line.split()])

D = zip(*data_)

if __name__ == '__main__':
    t = D[0]
    A = D[1]
    B = D[2]
    plot(t, A, 'bo')
    hold('on')
    plot(t, B, 'gX')
    xlabel('timestamp (unix)')
    ylabel('Station population')
    legend('Station 1','Station 2')
    title('Variance of Stations 1 and 2')
    show()
    savefig('2_stations_vs_time.png')

我已经测试过你的D值是否正确，例如D = [list(range(100)), list(range(10, 110)), list(range(20, 120))]。代码运行良好。