Skip to content
Snippets Groups Projects
Commit 00911818 authored by Terézia Slanináková's avatar Terézia Slanináková
Browse files

Fix for wrong export of trips.csv

parent 9fc33ae1
No related branches found
No related tags found
No related merge requests found
This diff is collapsed.
...@@ -78,7 +78,6 @@ def get_similar_cities(df, city_dict, cities, n_of_matches): ...@@ -78,7 +78,6 @@ def get_similar_cities(df, city_dict, cities, n_of_matches):
def load_data(folder="../data/"): def load_data(folder="../data/"):
df_trips = pd.read_csv(f"{folder}trips.csv", sep='\s*,\s*', encoding='utf-8') df_trips = pd.read_csv(f"{folder}trips.csv", sep='\s*,\s*', encoding='utf-8')
df_trips.city = pd.Series([ str(city).lower().replace(' ', '-') for city in df_trips.city.values]) df_trips.city = pd.Series([ str(city).lower().replace(' ', '-') for city in df_trips.city.values])
df_trips = df_trips.drop(["Unnamed: 0"], axis=1)
df_trips["trip_duration"] = pd.to_timedelta(df_trips["trip_duration"]) df_trips["trip_duration"] = pd.to_timedelta(df_trips["trip_duration"])
df_cities = pd.read_csv(f"{folder}cities_stats_full.csv", encoding='utf-8') df_cities = pd.read_csv(f"{folder}cities_stats_full.csv", encoding='utf-8')
return df_trips, df_cities return df_trips, df_cities
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment