swift_array_set_stats.py

import pandas as pd
from scipy.stats import ttest_ind 

with open('array_set.txt', 'r') as file:
    data_text = file.read()

splitted = data_text.strip('[(').strip(')]').strip(')]\n').split('), (')

data_list = []
for record in splitted:
    numbers = record.split(', ')
    formatted_numbers = [numbers[0], int(numbers[1]), float(numbers[2]), float(numbers[3])]
    data_list.append(formatted_numbers)

df = pd.DataFrame(data_list, columns =['experiment', 'trial', 'array_time', 'set_time'])

experiments = df['experiment'].unique()

results = []
for experiment in experiments:
    data1 = df[(df['experiment']==experiment) & (df['trial']>2)]
    data2 = df[(df['experiment']==experiment) & (df['trial']>2)]
    t_value, p_value =  ttest_ind(data1['array_time'], data2['set_time'])
    results.append((experiment, data1['array_time'].mean(), data2['set_time'].mean(), t_value, p_value))

print(*results, sep='\n')