FFT

import pylab as pl
import numpy as np

class FFT :
    def __init__(self, file='data.csv'):
        self.file=file
        self.readFile()

    def readFile(self) :
        xy = np.loadtxt(self.file, delimiter=",", skiprows=1).transpose()

        self.a = xy[4]
        self.u = xy[5]
        
    def U(self, angle):
        return self.u[min(range(len(self.a)), key=lambda x: abs(self.a[x] - angle))]