# ---
# jupyter:
#   jupytext:
#     text_representation:
#       extension: .py
#       format_name: light
#       format_version: '1.5'
#       jupytext_version: 1.14.4
#   kernelspec:
#     display_name: Python 3.6
#     language: python
#     name: py3
# ---

import pandas as pd
import numpy as np

# +
# df = pd.read_csv('NGSIM.csv')
# df[(df.Location == 'i-80')].to_csv('i_80.csv')
# df[(df.Location != 'i-80')].to_csv('non_i_80.csv')
# -

df = pd.read_csv('i_80.csv')
df = df.sort_values(by='Global_Time')
df[0:len(df)//3].to_csv('0_15_min.csv')
df[0:10000].to_csv('sample.csv')

df = pd.read_csv('sample.csv')

df = df[['Vehicle_ID', 'Frame_ID', 'Local_X', 'Local_Y','Lane_ID', 'v_length', 'v_Width', 'v_Class']]

df.iloc[0].Frame_ID



StarcoderdataJupyterScriptsDedupFiltered