diff --git a/labs06/task02.py b/labs06/task02.py index 87b38e4..013bfa6 100755 --- a/labs06/task02.py +++ b/labs06/task02.py @@ -1,16 +1,19 @@ import pandas as pd +import numpy as np def wczytaj_dane(): - mieszkania = pd.read_csv('mieszkania.csv', - index_col='Id') - return (mieszkania) + mieszkania = pd.read_csv('mieszkania.csv', + encoding='utf-8', + sep=',', + index_col='Id') + return pd.DataFrame(mieszkania) -def most_common_room_number(dane): - pass +def most_common_room_number (dane): + return dane['Rooms'].value_counts().head() def cheapest_flats(dane, n): - pass + return dane.sort_values('Expected').head(n) def find_borough(desc): dzielnice = ['Stare Miasto', @@ -21,14 +24,21 @@ def find_borough(desc): 'Winogrady', 'Miłostowo', 'Dębiec'] - pass - + for dzielnica in dzielnice: + if dzielnica in desc: + return dzielnica + return 'Inne' def add_borough(dane): - pass + dane['Borough'] = dane['Location'].apply(find_borough) + return dane def write_plot(dane, filename): - pass + data = add_borough(dane) + plot_data = data['Borough'].value_counts() + plot_data.plot( Kind = 'bar',title='Mieszkania z podziałem na dzielnice') + figure_data = plot_data.get_figure() + figure_data.savefig ( filename ) def mean_price(dane, room_number): pass @@ -43,14 +53,12 @@ def main(): dane = wczytaj_dane() print(dane[:5]) - print("Najpopularniejsza liczba pokoi w mieszkaniu to: {}" - .format(most_common_room_number(dane))) + print("Najpopularniejsza liczba pokoi w mieszkaniu to: {}".format ( most_common_room_number ( dane ) )) - print("{} to najłądniejsza dzielnica w Poznaniu." - .format(find_borough("Grunwald i Jeżyce")))) + print("Najtańsze mieszkania, to: {}".format(cheapest_flats(dane, 5))) + + print("{} to najłądniejsza dzielnica w Poznaniu.".format(find_borough("Grunwald i Jeżyce"))) - print("Średnia cena mieszkania 3-pokojowego, to: {}" - .format(mean_price(dane, 3))) if __name__ == "__main__": main() \ No newline at end of file