plots.py

import pandas as pd
import polyline

def get_segment(df , lId):
  df = df[df['id'].isin(lId)]
  df["polyline"] = df["map_pol"].apply(polyline.decode)
  return df["polyline"].tolist()