import numpy as np
import matplotlib.pyplot as plt

def plot():
    files = [
        "data/probability_deviation_no_slits.txt",
        "data/probability_deviation_slits.txt",
    ]
    for file in files:
        with open(file) as f:
            lines = f.readlines();
            x = []
            arr = []
            for line in lines:
                tmp = line.strip().split("\t")
                x.append(float(tmp[0]))
                arr.append(float(tmp[1]))

            plt.plot(x,arr)

        plt.savefig("latex/images/probability_deviation.pdf")

if __name__ == "__main__":
    plot()