Matplotlib初体验

为一个客户做了关于每个差异otu在时间点上变化的折线图，使用python第一次做批量作图的程序，虽然是很简单的折线图，但是也是第一次使用matplotlib的纪念。

ps：在第一个脚本上做了点小的改动，加上了分类信息作为图的标题，加上网格便于对照y轴丰度值，x轴的名称更加接近样品的实际名称。

from __future__ import division
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.ticker import MultipleLocator, FormatStrFormatter
import sys

def main():
    file = sys.argv[1]
    x = np.array([0,6,12,18])
    mean_file = ['../c-%dm_vs_cf-%dm.mean.profile'%(t,t) for t in [0,6,12,18]]
    with open(file) as f:
        f.next()
        for line in f:
            C = []
            CF = []
            tabs = line.strip().split('	')
            otu_name = tabs.pop(0)
            for file in mean_file:
                (C_temp,CF_temp) = getMeanProfile(otu_name,file)
                C.append(float(C_temp))
                CF.append(float(CF_temp))
            C = np.array(C)
            CF = np.array(CF)
            tax_name = tax[otu_name]
            do_plot(tax_name,otu_name,x,C,CF)
        f.close()

def getMeanProfile(otu_name,file):
    handle = open(file)
    handle.next()
    for line in handle:
        tabs = line.strip().split('	')
        if otu_name != tabs[0]:
            continue
        return( (tabs[1],tabs[2]) )
    handle.close()

def do_plot(title,file_name,x,y1,y2):
    plt.figure(figsize=(10,6))
    ax = plt.subplot(111)
    plt.plot(x,y1,label="C",color="red",linewidth=2)
    plt.plot(x,y2,label="CF",color="blue",linewidth=2)
    xmajorLocator = MultipleLocator(6)
    xmajorFormatter = FormatStrFormatter('%dm')
    yRange = ( max(np.max(y1),np.max(y2)) - min(np.min(y1),np.min(y2)) )
    ymajorLocator = MultipleLocator(yRange/10)
    yminorLocator = MultipleLocator(yRange/40)
    ax.xaxis.set_major_locator(xmajorLocator)
    ax.xaxis.set_major_formatter(xmajorFormatter)
    ax.yaxis.set_major_locator(ymajorLocator)
    ax.yaxis.set_minor_locator(yminorLocator)
    ax.xaxis.grid(True,which='major')
    ax.yaxis.grid(True,which='minor')
    plt.xlabel("Month(s)")
    plt.ylabel("mean_profile")
    plt.title(title)
    plt.legend()
    plt.savefig("%s.png"%file_name,dpi=80)

def getTax():
    for line in open('../tax.txt'):
        tabs = line.strip().split('	')
    for line in open('../tax.txt'):
        tabs = line.strip().split('	')
        tax[tabs[0]] = tabs[1].split(';')[-1]

if __name__ == '__main__':
    tax = {}
    getTax()
    main()