File size: 256 Bytes
73e9c25
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
import pandas as pd

allowed = ["CA 134", "CA 170","I 605"]

df = pd.read_csv("/Users/amitomer/Desktop/Personal/University/deep_learning/TrafCast/old_data/split_df_v2.csv")

df = df[df["road_name"].isin(allowed)]

df.to_csv("weights_test.csv", index=False)