眼动和脑波相关性分析后数据整理和查看

import pandas as pd
import numpy as np
import os
import matplotlib.pyplot as plt
def mkdir(path_3):
    folder = os.path.exists(path_3)
    if not folder:  # 判断是否存在文件夹如果不存在则创建为文件夹
        os.makedirs(path_3)  # makedirs 创建文件时如果路径不存在会创建这个路径
        print("---  new folder...  ---")
        print("---  OK  ---")
    else:
        print("---  There is this folder!  ---")

def read_file():
    gaze_data = pd.DataFrame(pd.read_csv(gaze_path + gaze_csv, encoding="utf-8", engine='python'))
    Clpd = np.array(gaze_data['Clpd'])
    print(Clpd)
    delta = Clpd[0]
    theta = Clpd[1]
    alpha = Clpd[2]
    beta = Clpd[3]
    return delta,theta,alpha,beta



path = './eeg_gaze_synchronization/csv/'

video_list = os.listdir(path)
for v in range(len(video_list)):
    Video = video_list[v]
    mid_path = path + Video +'/'
    channel_list = os.listdir(mid_path)
    for ch in range(len(channel_list)):
        Channel = channel_list[ch]
        gaze_path = mid_path + Channel +'/'
        gaze_list  = os.listdir(gaze_path)
        path_out = './eeg_gaze_synchronization/Correlation/csv/'+str(Video) +'/'
        path_figure = './eeg_gaze_synchronization/Correlation/figure/' + str(Video) + '/'
        mkdir(path_out);mkdir(path_figure)
        Delta = [];Theta = [];Alpha =[];Beta = []
        for g in range(len(gaze_list)):
            gaze_csv = gaze_list[g]
            delta, theta, alpha, beta = read_file()
            Delta.append(delta)
            Theta.append(theta)
            Alpha.append(alpha)
            Beta.append(beta)
        eeg_data = {'Delta': Delta, 'Theta': Theta, 'Alpha': Alpha,
                    'Beta': Beta}
        result = pd.DataFrame(eeg_data, columns=['Delta', 'Theta', 'Alpha', 'Beta'])
        result.to_csv(path_out + str(Channel) + '.csv')
        figsize = 14, 9
        figure, ax = plt.subplots(figsize=figsize)
        plt.subplot(2, 2, 1)
        plt.plot(Delta)
        plt.subplot(2, 2, 2)
        plt.plot(Theta)
        plt.subplot(2, 2, 3)
        plt.plot(Alpha)
        plt.subplot(2, 2, 4)
        plt.plot(Beta)
        plt.savefig(path_figure + Channel + '.png')
        Delta.clear();Theta.clear();Alpha.clear();Beta.clear()