In [1]:
import pandas as pd
import datetime
In [2]:
data = pd.read_csv('FinalMerge.csv',encoding='latin1')
In [3]:
# index of released date col
index = data.columns.get_loc("Released")
#change date data to timestamp
date_list = pd.to_datetime(data["Released"])
In [4]:
# released date is weekend of not
weekend_list = []
for each in date_list:
day_ofweek = each.dayofweek
if day_ofweek >= 4 and day_ofweek <= 6:
tag = 1
else:
tag = 0
weekend_list.append(tag)
In [5]:
# released date is on dump months
dumpmonth_list = []
for each in date_list:
month = each.month
if month == 1 or month == 2 or month == 8 or month ==9:
tag = 1
else:
tag = 0
dumpmonth_list.append(tag)
In [6]:
data.insert(loc=index+1,column = "Released on weekend",value=weekend_list)
data.insert(loc=index+2,column = "Released on dump month",value=dumpmonth_list)
data.to_csv("FinalMerge_updateson_releaseddate.csv")