import csv

with open ("python_final_project_2022-07-27.csv", newline= "") as csvfile:
    csvfile = csv.reader(csvfile)
    headings = next(csvfile)
    pos_con = headings.index("player.condition")
    pos_choice = headings.index("player.choice_val")

    con1_res = []
    con2_res = []
    con3_res = []

    for line in csvfile:
        if line[pos_con] == "1":
            con1_res.append(int(line[pos_choice]))
        elif line[pos_con] == "2":
            con2_res.append(int(line[pos_choice]))
        else:
            con3_res.append(int(line[pos_choice]))

    print(sum(con1_res)/len(con1_res))
    print(sum(con2_res)/len(con2_res))
    print(sum(con3_res)/len(con3_res))

with open ("experiment_data.csv", "w", newline= "") as file:
    writer = csv.writer(file)
    writer.writerow(["Condition", "Risk Rate"])
    writer.writerow(["All_Forgone", (sum(con1_res)/len(con1_res))])
    writer.writerow(["Only_Better_Forgone", (sum(con2_res)/len(con2_res))])
    writer.writerow(["Only_Worse_Forgone", (sum(con3_res)/len(con3_res))])