在Python中对同一目录中的多个文件进行Forloop

2024-09-28 03:17:20 发布

您现在位置:Python中文网/ 问答频道 /正文

所以我已经试着在这里检查关于(几乎)同一主题的其他问题,但是我没有找到解决我问题的方法。你知道吗

基本上,我有一段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")

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


Tags: 文件csv代码indffordatatime
1条回答
网友
1楼 · 发布于 2024-09-28 03:17:20

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

#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)

相关问题 更多 >

    热门问题