У меня есть данные с отметкой времени в UTC. Я хотел бы преобразовать часовой пояс этой метки времени в "US/Pacific" и добавить ее как иерархический индекс в pandas DataFrame. Я смог преобразовать метку времени как индекс, но он теряет форматирование часового пояса, когда я пытаюсь добавить его обратно в DataFrame, либо в виде столбца, либо как индекс.
>>> import pandas as pd
>>> dat = pd.DataFrame({'label':['a', 'a', 'a', 'b', 'b', 'b'], 'datetime':['2011-07-19 07:00:00', '2011-07-19 08:00:00', '2011-07-19 09:00:00', '2011-07-19 07:00:00', '2011-07-19 08:00:00', '2011-07-19 09:00:00'], 'value':range(6)})
>>> dat.dtypes
#datetime object
#label object
#value int64
#dtype: object
Теперь, если я попытаюсь преобразовать серию непосредственно, я столкнулся с ошибкой.
>>> times = pd.to_datetime(dat['datetime'])
>>> times.tz_localize('UTC')
#Traceback (most recent call last):
# File "<stdin>", line 1, in <module>
# File "/Users/erikshilts/workspace/schedule-detection/python/pysched/env/lib/python2.7/site-packages/pandas/core/series.py", line 3170, in tz_localize
# raise Exception('Cannot tz-localize non-time series')
#Exception: Cannot tz-localize non-time series
Если я конвертирую его в индекс, я могу манипулировать им как таймсерий. Обратите внимание, что индекс теперь имеет тихоокеанский часовой пояс.
>>> times_index = pd.Index(times)
>>> times_index_pacific = times_index.tz_localize('UTC').tz_convert('US/Pacific')
>>> times_index_pacific
#<class 'pandas.tseries.index.DatetimeIndex'>
#[2011-07-19 00:00:00, ..., 2011-07-19 02:00:00]
#Length: 6, Freq: None, Timezone: US/Pacific
Однако теперь я сталкиваюсь с проблемами, добавляющими индекс обратно в dataframe, поскольку он теряет свое форматирование в часовом поясе:
>>> dat_index = dat.set_index([dat['label'], times_index_pacific])
>>> dat_index
# datetime label value
#label
#a 2011-07-19 07:00:00 2011-07-19 07:00:00 a 0
# 2011-07-19 08:00:00 2011-07-19 08:00:00 a 1
# 2011-07-19 09:00:00 2011-07-19 09:00:00 a 2
#b 2011-07-19 07:00:00 2011-07-19 07:00:00 b 3
# 2011-07-19 08:00:00 2011-07-19 08:00:00 b 4
# 2011-07-19 09:00:00 2011-07-19 09:00:00 b 5
Вы заметите, что индекс возвращается в часовой пояс UTC вместо измененного часового пояса Тихого океана.
Как изменить часовой пояс и добавить его в качестве индекса в DataFrame?