import pandas as pd
df = pd.read_csv('22_may_con.csv',sep =',')
#print(df["peak_hour"][0])
a = df["centroid_lat"]
b = df["centroid_lon"]
c = df["car_hours"]
d = df["peak_hour"]
#print(a,b,c,d)
l1,l2 = [],[]
a1,b1,c1,d1 = [],[],[],[]
a2,b2,c2,d2 = [],[],[],[]
for i in range(len(d)):
	if d[i] ==0:
		a1.append(a[i])
		b1.append(b[i])
		c1.append(c[i])
		d1.append(d[i])
	elif d[i] ==1:
		a2.append(a[i])
		b2.append(b[i])
		c2.append(c[i])
		d2.append(d[i])
	else:
		a1.append(a[i])
		b1.append(b[i])
		c1.append(c[i])
		d1.append(d[i])
		a2.append(a[i])
		b2.append(b[i])
		c2.append(c[i])
		d2.append(d[i])
				
l1.append(a1)
l1.append(b1)
l1.append(c1)
l1.append(d1)
#print(l1)
#l2.append(a2,b2,c2,d2)
#df1 = pd.DataFrame(a1,b1,columns = ["centroid_lat","centroid_lon"])
data_1 ={"centroid_lat": a1, "centroid_lon": b1,"car_hours":c1,"peak_hour":d1}
data_2 ={"centroid_lat": a2, "centroid_lon": b2,"car_hours":c2,"peak_hour":d2}
df1 = pd.DataFrame(data_1)
df1.to_csv("22_may_CG.csv")
df2 = pd.DataFrame(data_2)
df2.to_csv("22_may_IC.csv")
#df1.columns = ["centroid_lat","centroid_lon","car_hours","peak_hour"]
#print(df1)