def writeToCSV(numOfTeams):
# header list
headerList = [key.name for key in CSVReader.DataColumnsFromCSV]
appendedColumnForTeamName = ["TeamName"]
headerList = appendedColumnForTeamName + headerList
headerList = dataCustomizer(headerList)
dataListToCSV = []
for team in TeamSorter.finalTeamsDictionary:
for names in TeamSorter.finalTeamsDictionary[team]:
playerInformation = [
team
] + CSVReader.getDictionaryAllPlayerInformation()[names]
dataListToCSV.append(dataCustomizer(playerInformation))
dataListToCSV.append([])
print(dataListToCSV)
preprocessedDataToWrite = pd.DataFrame(dataListToCSV,
columns=headerList)
preprocessedDataToWrite.to_csv('generatedTeams.csv', index=False)
preprocessedDataToWrite.to_csv('generatedTeamsBACKUP.csv',
index=False)
def __sortPlayersByGender(optionToSort, isRandom):
unsortedGenderDictionaryWithSortedSkills = TeamSorter.__getDictionaryOfPlayersSortBySkill(
optionToSort, isRandom)
informationDictionary = CSVReader.getDictionaryAllPlayerInformation()
for player in unsortedGenderDictionaryWithSortedSkills:
# Assigns males
if informationDictionary[player][
CSVReader.DataColumnsFromCSV.gender.value -
1].lower() == 'male':
TeamSorter.malePlayersArraySortedBySkills.append(player)
# Assigns females
else:
TeamSorter.femalePlayersArraySortedBySkills.append(player)
