diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-05 13:11:18 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-05 13:11:33 -0400 |
| commit | 712035b88be1816d3fbd58ce69ae6464767c780e (patch) | |
| tree | 85a56482c508e10b946e6501544e86f62cab53a7 /transformers.py | |
| parent | 66159d9fce0129116e82e74cf3eb1d9e048b253d (diff) | |
| download | taxi-712035b88be1816d3fbd58ce69ae6464767c780e.tar.gz taxi-712035b88be1816d3fbd58ce69ae6464767c780e.zip | |
Add day type and taxi id
Diffstat (limited to 'transformers.py')
| -rw-r--r-- | transformers.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/transformers.py b/transformers.py index d6ed611..6ee0df1 100644 --- a/transformers.py +++ b/transformers.py @@ -107,7 +107,8 @@ class TaxiAddDateTime(Transformer): data = next(self.child_epoch_iterator) ts = data[self.id_timestamp] date = datetime.datetime.utcfromtimestamp(ts) - info = (date.isocalendar()[1] - 1, date.weekday(), date.hour * 4 + date.minute / 15) + yearweek = date.isocalendar()[1] - 1 + info = ((51 if yearweek == 52 else yearweek), date.weekday(), date.hour * 4 + date.minute / 15) return data + info class TaxiExcludeTrips(Transformer): @@ -122,4 +123,3 @@ class TaxiExcludeTrips(Transformer): if not data[self.id_trip_id] in self.exclude: break return data - |