crowdedness_create_features

crowdedness_create_features.py

features=[]
for rn in timedata.ritnumber.unique():
    total_trip = timedata[timedata.ritnumber == rn].dropna()
    for sq in total_trip.sequence.unique():
        trip = total_trip[total_trip.sequence == sq]
        tripfirst = trip.iloc[0]
        triplast = trip.iloc[-1]
        name = str(tripfirst["ritnumber"]) + " : " + tripfirst["stop_code"] + \
              " - " + triplast["stop_code"]
        ls = LineString(trip[['lon', 'lat', 'elevation', 'timestamp']]. \
            to_numpy().tolist())
        features.append(
            Feature(geometry=ls ,
                properties=dict(label=str(tripfirst["ritnumber"]),
                                name=name,
                                capacity=str(tripfirst["seats"]),
                                passengers=str(tripfirst["passengers"]),
                                crowding=str(tripfirst["classification"])))    
        )

with open('trainpag.geojson', 'w') as outfile:
        dump(FeatureCollection(features), outfile)