From f3ff46f3497433d4eb830a49d7f0875150a76b7c Mon Sep 17 00:00:00 2001 From: s444510 Date: Sat, 25 Mar 2023 16:02:06 +0100 Subject: [PATCH] add main.py --- main.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 main.py diff --git a/main.py b/main.py new file mode 100644 index 0000000..aedd2fa --- /dev/null +++ b/main.py @@ -0,0 +1,28 @@ +import jupyter +import numpy as np +import pandas as pd +from sklearn.model_selection import train_test_split +import kaggle +import pandas +import os + +path_to_data = './data' + + +def download_data(): + kaggle.api.authenticate() + kaggle.api.dataset_download_files('thedevastator/airbnb-prices-in-european-cities', path=path_to_data, unzip=True) + + +def read_data_from_file(filename): + return pandas.read_csv(filename) + + +if __name__ == '__main__': + if not os.path.isdir('data'): + download_data() + df = read_data_from_file(path_to_data + '/barcelona_weekends.csv') + print(df.head()) + + +