LockInDataPlottingBatchProcess.py

# -*- coding: utf-8 -*-
"""
Created on Tue Apr  4 18:20:16 2017

@author: dwi
"""

import numpy as np
import pandas as pd
import matplotlib.pyplot as pl
import cmocean
from matplotlib import ticker
import glob

files = glob.glob(r"/home/dwi/ownCloud/Anlabshared/Dwi/Projects/Spin-waveHeatConveyer/LockinThermographyData/20170222/AllCSVAndFigures/*ampl.csv")

for f in files:
    Data = pd.read_csv(f, usecols = range(1, 640) , skiprows = 4)
    pl.close() #Closing previous figure
    I = np.array(Data) #converting data into array
    x = np.arange(0, 640, 1) #X-axis
    y = np.arange(0, 512, 1) #Y-axis
    X, Y = np.meshgrid(x, y) #Grid of X and Y-axis
    pl.rc('text', usetex=True)
    pl.rc('font', family='serif', size = 20)
    pl.xlim(0, 640)
    pl.ylim(0, 512)
    pl.gca().invert_yaxis() #Inverting Y axis
    #pl.imshow(I, cmap = 'jet')
    pl.pcolormesh(X, Y, I, cmap = cmocean.cm.thermal, vmax = 50.0, vmin = 0.0) #Amplitude
    #pl.pcolormesh(X, Y, I, cmap = 'hsv_r', vmax = 360.0, vmin = 0.0) #Phase
    cb = pl.colorbar(label = 'A (mK)')
    #cb = pl.colorbar(label = r'\phi (deg)')
    tick_locator = ticker.MaxNLocator(nbins=6)
    cb.locator = tick_locator
    cb.update_ticks()
    frame1 = pl.gca()
    frame1.axes.get_xaxis().set_visible(False)
    frame1.axes.get_yaxis().set_visible(False)
    pl.savefig(f[:-4] + '.png', format = 'png') #Save Figure and delete file extension
    
    #pl.show()

#print (I)