我在大熊猫0.14.1上。假设我需要使用时区在分层索引中通过两个时间戳索引数据。将结果DataFrame保存到hdf5时,我似乎失去了时区感知功能:
import pandas as pd
dti1 = pd.DatetimeIndex(start=pd.Timestamp('20000101'), end=pd.Timestamp('20000102'), freq='D', tz='EST5EDT')
dti2 = pd.DatetimeIndex(start=pd.Timestamp('20000102'), end=pd.Timestamp('20000103'), freq='D', tz='EST5EDT')
mux = pd.MultiIndex.from_arrays([dti1, dti2])
df = pd.DataFrame(0, index=mux, columns=['a'])
此处df
有时区:
a
2000-01-01 00:00:00-05:00 2000-01-02 00:00:00-05:00 0
2000-01-02 00:00:00-05:00 2000-01-03 00:00:00-05:00 0
保存并加载到hdf5后,时区信息似乎消失了:
df.to_hdf('/tmp/my.h5', 'data')
pd.read_hdf('/tmp/my.h5', 'data')
结果:
a
2000-01-01 05:00:00 2000-01-02 05:00:00 0
2000-01-02 05:00:00 2000-01-03 05:00:00 0
我想知道是否有一个好的解决方法以及这是否是一个知道错误。
答案 0 :(得分:3)
使用多索引时,fixed
格式不支持此功能。我想应该提高我没想到的实施。这是跟踪this
请参阅full-hdf5-interface docs here
In [11]: pd.read_hdf('/tmp/my.h5', 'data').index.levels[0]
Out[11]:
<class 'pandas.tseries.index.DatetimeIndex'>
[2000-01-01 05:00:00, 2000-01-02 05:00:00]
Length: 2, Freq: None, Timezone: None
但如果指定table
格式,则可以使用。
In [13]: df.to_hdf('/tmp/my.h5', 'data2', format='table')
In [14]: pd.read_hdf('/tmp/my.h5', 'data2')
Out[14]:
a
2000-01-01 00:00:00-05:00 2000-01-02 00:00:00-05:00 0
2000-01-02 00:00:00-05:00 2000-01-03 00:00:00-05:00 0
In [15]: pd.read_hdf('/tmp/my.h5', 'data2').index.levels[0]
Out[15]:
<class 'pandas.tseries.index.DatetimeIndex'>
[2000-01-01 00:00:00-05:00, 2000-01-02 00:00:00-05:00]
Length: 2, Freq: None, Timezone: EST5EDT
In [16]: pd.read_hdf('/tmp/my.h5', 'data2').index.levels[1]
Out[16]:
<class 'pandas.tseries.index.DatetimeIndex'>
[2000-01-02 00:00:00-05:00, 2000-01-03 00:00:00-05:00]
Length: 2, Freq: None, Timezone: EST5EDT