Some spaghetti code I threw together in order to sort survey data for a school project in sociology class.

survey_sorter.py

"""
Some spaghetti code I threw together in order to sort
survey data for a school project.
"""


import xlrd
import xlwt


excel_file = xlrd.open_workbook(filename=f'input/raw_data.xlsx')
excel_sheet = excel_file.sheet_by_index(0)
wb = xlwt.Workbook()


def sorter(sheetname, rownumber):
    drinkers = 0
    non_drinkers = 0
    sheet = wb.add_sheet(sheetname)
    for row in range(excel_sheet.nrows):
        x = excel_sheet.cell_value(row, rownumber)
        if x != sheetname:
            continue
        drinkstatus = excel_sheet.cell_value(row, 3)
        if drinkstatus == 'Ja':
            drinkers += 1
        elif drinkstatus == 'Nei':
            non_drinkers += 1

    sheet.write(0, 0, 'drikker')
    sheet.write(0, 1, drinkers)
    sheet.write(1, 0, 'drikker ikke')
    sheet.write(1, 1, non_drinkers)
    wb.save('output/sorted.xlsx')


sorter('Gutt', 1)
sorter('Jente', 1)
sorter('VG1', 2)
sorter('VG2', 2)
sorter('VG3', 2)