From d8c5829e4461cf50ba3ee0d92a6cbc6957ca9d0a Mon Sep 17 00:00:00 2001 From: Sijia Wang Date: Tue, 21 May 2024 19:16:40 -0400 Subject: [PATCH] `joint_tour_frequency_composition` column to categorical (#866) * joint tour freq composition column * convert jtfc alt to int, auto downcast to efficient int type --- activitysim/abm/models/joint_tour_frequency_composition.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/activitysim/abm/models/joint_tour_frequency_composition.py b/activitysim/abm/models/joint_tour_frequency_composition.py index f3e4a7df8..553b280fe 100644 --- a/activitysim/abm/models/joint_tour_frequency_composition.py +++ b/activitysim/abm/models/joint_tour_frequency_composition.py @@ -189,8 +189,10 @@ def joint_tour_frequency_composition( # we expect there to be an alt with no tours - which we can use to backfill non-travelers no_tours_alt = 0 - households_merged["joint_tour_frequency_composition"] = ( - choices.reindex(households_merged.index).fillna(no_tours_alt).astype(str) + # keep memory usage down by downcasting + households_merged["joint_tour_frequency_composition"] = pd.to_numeric( + choices.reindex(households_merged.index).fillna(no_tours_alt), + downcast="integer", ) households_merged["num_hh_joint_tours"] = (