source: ETALON/CLIO/control/plot_file_average.py @ 691

# -*- coding: utf-8 -*-
"""
Created on Sun Mar  5 18:51:17 2017

Calculates the mean of several data set

@author: delerue
"""

import glob
import os
import numpy as np

global data_directory

from clio_constants import *

filemask="data_CLIO*.zoom_txt"
list_of_files = glob.glob(data_directory+filemask)

print repr(list_of_files)

#prepare the figures    
fig=plt.figure(9,figsize=(12, 8))

nfiles=0
for filename in list_of_files:
    print filename
    filedata=np.loadtxt(filename)
    if (nfiles==0):
        nchannels=np.size(filedata,0)
        ndatapoints=np.size(filedata,1)
        alldata=np.zeros((nchannels,ndatapoints))
    else:
        if (nchannels!=np.size(filedata,0)):
            print "Incorrect number of channels in file!"
        if (ndatapoints!=np.size(filedata,1)):
            print "Incorrect number of data points in file!"
    plt.figure(9)
    plt.plot(range(1,ndatapoints),filedata[1,range(1,ndatapoints)])
    print repr(alldata)
    alldata=alldata+filedata
    nfiles+=1
    
    print "nfiles "+str(nfiles)
    
 #prepare the figures    
fig=plt.figure(1,figsize=(12, 8))
plt.figure(1)
plt.plot(range(1,ndatapoints),alldata[1,range(1,ndatapoints)])