A utility function to localize a UTC timestamp DataFrame column

localize_tms.py

import pytz

def localize_datetime(input_df, timezone, tms_col):
    """
    Convert datetime column from UTC to another timezone.
    """
    tmz = pytz.timezone(timezone)
    df = input_df.copy()
    return (df.set_index(tms_col)
              .tz_localize(pytz.utc) # UTC time
              .tz_convert(tmz) # Timezone time
              .reset_index())