line_graph_with_control.py

import numpy as np
import matplotlib.pyplot as plt
import matplotlib
from sys import argv, exit

if len(argv) < 6:
    print("Usage: ./script WHICH AXIS_X AXIS_Y CONTROL FILE [FILE, FILE, ...]")
    print("WHICH = <column #>[,<column #>,...]")
    print("AXIS_X = <x axis label>")
    print("AXIS_Y = <y axis label>")
    exit(-1)

WHICH = [int(x) for x in argv[1].split(',')]
AXIS_X = argv[2]
AXIS_Y = argv[3]
CONTROL = argv[4]

data = {}

for data_file in argv[4:]:
    with open(data_file, 'r') as f:
        data[data_file] = []

        for line in f:
            if len(line.strip()) == 0:
                continue

            xs = [float(x) for x in line.split()]

            data[data_file].append(xs)

handles = []

for data_file in data:
    for i in WHICH:
        if i < len(data[data_file][0]):
            ys = [data[data_file][x][i] - data[data_file][0][i] - data[CONTROL][x][i] for x in range(len(data[data_file]))]
            line, = plt.plot(ys, label = data_file)
            handles.append(line)

#plt.xscale("log", basex = 2)
#plt.yscale(SCALE)

#plt.xlim(0, 512+32)
#plt.ylim(0, 600)

plt.xlabel(AXIS_X)
plt.ylabel(AXIS_Y)
plt.grid(True)

plt.legend(handles = handles)

font = {'family' : 'normal',
        'weight' : 'bold',
        'size'   : 22}

matplotlib.rc('font', **font)

plt.show()