Generate stations_mini.csv with suggestable stations only

tducret · Jun 1, 2019 · a3b60f6 · a3b60f6
1 parent bcdc5ca
commit a3b60f6
Show file tree

Hide file tree

Showing 2 changed files with 1,873 additions and 5,377 deletions.
diff --git a/build_station_index.py b/build_station_index.py
@@ -14,6 +14,6 @@
 csv_content = requests.get(_STATIONS_CSV_FILE).content
 df = pd.read_csv(io.StringIO(csv_content.decode('utf-8')),
                  sep=';', index_col=0, low_memory=False)
-
+df = df[df.is_suggestable == 't']
 df_mini = df.name.str.lower()
-df_mini.to_csv("stations_mini.csv", sep=';', encoding='utf-8')
+df_mini.to_csv("stations_mini.csv", sep=';', encoding='utf-8', header=False)