From 3d0e2d9889c54321f0beb4bf9c68d6bb7b3ceff0 Mon Sep 17 00:00:00 2001 From: pietrzakkuba Date: Mon, 20 Dec 2021 10:51:28 +0100 Subject: [PATCH] reading data from console --- main.py | 21 +++++++++++++++++++-- read_data.py | 4 ++-- 2 files changed, 21 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index 7cb98a6..25c9f94 100644 --- a/main.py +++ b/main.py @@ -207,8 +207,25 @@ def defineRules(): if __name__ == "__main__": initFuzzyMembershipFunctions() - leagues_data = get_leagues_data() - for league in leagues_data: + data_source = None + while not(data_source == '1' or data_source == '2'): + data_source = input('Wpisz 1 by wczytać dane z pliku lub 2 by wpisać dane z konsoli:\n') + if data_source == '1': + leagues_data = get_leagues_data('leagues.csv') + for league in leagues_data: + activateFuzzyMembershipFunctions(league) + result = defineRules() + print(f"{league['leagueName']} score is: {result}") + else: + league = {'leagueName': input('Podaj nazwę ligi:\n'), 'country': input('Podaj kraj ligi:\n'), + 'clubs': input('Podaj liczbę klubów grających w lidze:\n'), + 'goalsPerMatch': input('Podaj średnią liczbę goli na mecz:\n'), + 'avgMarketValueInMln': input('Podaj średnią wartośc klubu w lidze (w milionach):\n'), + 'totalMarketValueInMln': input('Podaj całkowitą wartość ligi (w milionach):\n'), + 'teamsInChampionsLeague': input('Podaj liczbę klubów grających w Lidze Mistrzów:\n'), + 'teamsInEuropaLeague': input('Podaj liczbę klubów grających w Lidze Europy:\n'), + 'teamsInConferenceLeague': input('Podaj liczbę klubów grających w Lidze Konferencji:\n'), + 'famousPlayers': input('Podaj liczbę sławnych piłkarzy:\n')} activateFuzzyMembershipFunctions(league) result = defineRules() print(f"{league['leagueName']} score is: {result}") diff --git a/read_data.py b/read_data.py index 02c07af..6158cb4 100644 --- a/read_data.py +++ b/read_data.py @@ -1,7 +1,7 @@ import pandas as pd -def get_leagues_data(): - df = pd.read_csv('leagues.csv', sep=";", header=0) +def get_leagues_data(filename): + df = pd.read_csv(filename, sep=";", header=0) return [row for _, row in df.iterrows()]