workbook = xlsxwriter.Workbook(filename)
row = 0
col = 0
for sheet_name in list_name:
worksheet = workbook.add_worksheet(sheet_name)
if sheet_name == 'bandit_task':
row_headers = ['participant_num', 'cond_num', 'cpt_iteration', 'reaction_times', 'feedbacks',
'pressed_left', 'age', 'gender']
worksheet.write_row(row, col, tuple(row_headers))
for i in range(0, len(reaction_times_experiment)):
row += 1
worksheet.write_row(row, col,
tuple((participant_number, cond_num, i, reaction_times_experiment[i], feedbacks[i],
pressed_lefts[i], age, gender)))
if sheet_name == 'rating':
row = 0
row_headers = ['initial_base_line', 'initial_rating_score', 'initial_reaction_time',
'final_base_line', 'final_rating_score', 'final_reaction_time']
worksheet.write_row(row, col, tuple(row_headers))
for i in range(0, len(initial_ratings)):
row += 1
worksheet.write_row(row, col,
tuple((initial_ratings[i][2], initial_ratings[i][0], initial_ratings[i][1],
final_ratings[i][2], final_ratings[i][0], final_ratings[i][1])))
workbook.close()
core.quit()
core.close()
def save_progress():
# Store info about the experiment session
expName = 'bandit_exp'
expInfo = {
'gui': {
'participant_number': participant_number,
'cond_num': cond_num,
'age': age,
'gender': gender
},
'date': data.getDateStr(),
'expName': expName
}
directory_name = 'Data cognitive task'
create_directory(directory_name)
filename = directory_name + '/' + '%s_%s_%s.xlsx' % (
participant_number, expName, expInfo['date'])
print('filepath is %s' % filename)
list_name = ['bandit_task', 'rating']
try:
workbook = xlsxwriter.Workbook(filename)
except Exception as e:
print('error while creating workbook')
write_to_log_error_file('error while creating workbook')
write_to_log_error_file(str(e))
exit()
try:
print('final_ratings.....')
global final_ratings
global final_ratings_temperament
global feedbacks
global initial_ratings
global initial_ratings_temperament
global pressed_lefts
global reaction_times_experiment
if not final_ratings:
final_ratings = [
init_to_999(initial_ratings[0]),
init_to_999(initial_ratings[1]),
init_to_999(initial_ratings[2])
]
if not final_ratings_temperament:
final_ratings_temperament = final_ratings
print(reaction_times_experiment)
print(feedbacks)
print(pressed_lefts)
print(initial_ratings)
print(final_ratings)
print(initial_ratings_temperament)
print(final_ratings_temperament)
print('********************')
feedbacks = remove_none(feedbacks)
pressed_lefts = remove_none(pressed_lefts)
initial_ratings = remove_none(initial_ratings)
final_ratings = remove_none(final_ratings)
initial_ratings_temperament = remove_none(initial_ratings_temperament)
final_ratings_temperament = remove_none(final_ratings_temperament)
print(reaction_times_experiment)
print(feedbacks)
print(pressed_lefts)
print(initial_ratings)
print(final_ratings)
print(initial_ratings_temperament)
print(final_ratings_temperament)
row = 0
col = 0
for sheet_name in list_name:
worksheet = workbook.add_worksheet(sheet_name)
if sheet_name == 'bandit_task':
row_headers = [
'participant_num', 'cond_num', 'cpt_iteration',
'reaction_times', 'feedbacks', 'pressed_left', 'age',
'gender'
]
worksheet.write_row(row, col, tuple(row_headers))
for i in range(0, len(reaction_times_experiment)):
row += 1
worksheet.write_row(
row, col,
tuple((participant_number, cond_num, i,
reaction_times_experiment[i], feedbacks[i],
pressed_lefts[i], age, gender)))
if sheet_name == 'rating':
row = 0
row_headers = [
'initial_base_line', 'initial_rating_score',
'initial_reaction_time', 'final_base_line',
'final_rating_score', 'final_reaction_time',
'initial_base_line_temperament',
'initial_rating_score_temperament',
'initial_reaction_time_temperament',
'final_base_line_temperament',
'final_rating_score_temperament',
'final_reaction_time_temperament'
]
worksheet.write_row(row, col, tuple(row_headers))
for i in range(0, len(initial_ratings)):
row += 1
print('i', i)
print(row, col)
print(initial_ratings[i][2], initial_ratings[i][0],
initial_ratings[i][1])
print('****')
print(final_ratings)
print(final_ratings[i])
print(final_ratings[i][2], final_ratings[i][0],
final_ratings[i][1])
print('****')
print(initial_ratings_temperament[i][2],
initial_ratings_temperament[i][0],
initial_ratings_temperament[i][1])
print('****')
print(final_ratings_temperament[i][2],
final_ratings_temperament[i][0],
final_ratings_temperament[i][1])
print('****')
print('****')
t = tuple((initial_ratings[i][2], initial_ratings[i][0],
initial_ratings[i][1], final_ratings[i][2],
final_ratings[i][0], final_ratings[i][1],
initial_ratings_temperament[i][2],
initial_ratings_temperament[i][0],
initial_ratings_temperament[i][1],
final_ratings_temperament[i][2],
final_ratings_temperament[i][0],
final_ratings_temperament[i][1]))
print(t)
worksheet.write_row(row, col, t)
except Exception as e:
print('error while writing in workbook')
write_to_log_error_file('error while writing in workbook')
write_to_log_error_file(str(e))
exit()
exit()
workbook.close()
core.quit()
core.close()
