Newer
Older
StressAnalysis / make_data.py
import pandas as pd

data_dir = r"D:\sync\working\StressFiles" + "\\"
target_list = ["SmTIAS_WebApp","HandyTCC_WebApp","SmTIAS_PhoneApp","HandyTCC_PhoneApp"]
survey = pd.read_csv(data_dir + 'survey_data_error_removed.csv')

def make_data(target):
    type = target.split('_')[1]
    type = type.replace('App', '')
    print(target)
    deeptias = pd.read_csv(data_dir + f'DeepTIAS_{type}.csv')

    merged = pd.merge(survey, deeptias, left_on=target, right_on='name', how='inner')
    merged.to_csv(f'Merged_{target}.csv', index=False)

    print(merged.head(3))

if __name__ == "__main__":
    for target in target_list:
        make_data(target)