For循环遍历Python中同一目录中的多个文件

时间:2018-04-24 12:34:17

标签: python for-loop

所以我已经尝试在这里查看关于(几乎)相同主题的其他问题,但是我没有找到解决我问题的东西。

基本上,我在Python中有一段代码试图将文件作为数据框打开并执行一些眼动追踪功能(PyGaze)。我有1000个文件需要分析,并希望创建一个for循环来自动执行所有文件的代码。

代码如下:

os.chdir("/Users/Documents//Analyse/Eye movements/Python - Eye Analyse")

directory = '/Users/Documents/Analyse/Eye movements/R - Filtering Data/Filtered_data/Filtered_data_test'

for files in glob.glob(os.path.join(directory,"*.csv")):
    #Downloas csv, plot 
    df = pd.read_csv(files, parse_dates = True)
    #Plot raw data
    plt.plot(df['eye_x'],df['eye_y'], 'ro', c="red")
    plt.ylim([0,1080])
    plt.xlim([0,1920])
    #Fixation analysis 
    from detectors import fixation_detection 
    fixations_data = fixation_detection(df['eye_x'],df['eye_y'], df['time'],maxdist=25, mindur=100)
    Efix_data = fixations_data[1]
    numb_fixations = len(Efix_data) #number of fixations
    fixation_start = [i[0] for i in Efix_data] 
    fixation_stop = [i[1] for i in Efix_data] 
    fixation = {'start' : fixation_start, 'stop': fixation_stop}   
    fixation_frame = pd.DataFrame(data=fixation)
    fixation_frame['difference'] = fixation_frame['stop'] - fixation_frame['start']
    mean_fixation_time = fixation_frame['difference'].mean() #mean fixation time 
    final = {'number_fixations' : [numb_fixations], 'mean_fixation_time': [mean_fixation_time]} 
    final_frame = pd.DataFrame(data=final)
    #write everything in one document
    final_frame.to_csv("/Users/Documents/Analyse/Eye movements/final_data.csv")

代码正在运行(没有错误),但是:它只运行第一个文件。不会为文件夹/目录中的其他文件运行代码。 我不知道我的错误在哪里?

1 个答案:

答案 0 :(得分:0)

您的输出文件名是常量,因此它会在for循环的每次迭代中被覆盖。请尝试以下操作,而不是最后一行,而是以“追加”模式打开文件:

#write everything in one document
with open("/Users/Documents/Analyse/Eye movements/final_data.csv", "a") as f:
    final_frame.to_csv(f, header=False)