Team-L/beforeTrain.py at master · shellwayne01/Team-L · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import pandas as pd
import folium

libs = pd.read_csv('train.csv')

coords = []
popups = []
icons = []
mapVor = folium.Map(location=[40.75, -73.9],tiles="Cartodb Positron",zoom_start=13)

for index, row in libs.iterrows():
    if (row['Line'] == "G" or row['Line'] == "L" or row['Line'] == "M"):
        lat = row['Latitude']
        lon = row['Longitude']
        name = row['Line'] + ' train at ' + row['Station']
        coords.append([lat,lon])

        if(row['Line'] == "G"):
             folium.Marker([lat,lon],popup = name, icon = folium.Icon(color='green')).add_to(mapVor)
        elif(row['Line'] == "M"):
             folium.Marker([lat,lon],popup = name, icon = folium.Icon(color='orange')).add_to(mapVor)
        elif(row['Line'] == "L"):
             folium.Marker([lat,lon],popup = name, icon = folium.Icon(color='gray')).add_to(mapVor)

mapVor.save(outfile='beforeTrain.html')