python - 存储按日期索引的对象的最佳数据结构是什么？

Question

我有一些与日期相关的对象。我希望能够获取给定日期范围以及单个日期的对象。python 中执行此操作的最佳数据结构是什么？

我相信在某些语言中，这可以使用“曲线”来实现。

Answer 1

你可以使用 pandas DataFrame :

import pandas as pd, numpy as np

df = pd.DataFrame(np.random.random((10,1)))
df['date'] = pd.date_range('2016-7-1', periods=10, freq='D')

对于日期范围:

>>> df[(df['date'] > '2016-7-3') & (df['date'] <= '2016-7-10')]
          0       date
3  0.322654 2016-07-04
4  0.360684 2016-07-05
5  0.298821 2016-07-06
6  0.292097 2016-07-07
7  0.052085 2016-07-08
8  0.620535 2016-07-09
9  0.902022 2016-07-10

对于给定的日期:

>>> df[df['date'] == '2016-7-3']
          0       date
2  0.191553 2016-07-03