add main.py
This commit is contained in:
parent
606666da7b
commit
f3ff46f349
28
main.py
Normal file
28
main.py
Normal file
@ -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())
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user