python-2.7 - Pandas:将 'crop' 作为大型数据帧仅存储到前 1000 天的最佳方法是什么？

Question

我有一个数据框，其中索引由日期时间组成。我还有一个锚定日期，并且我知道我只希望第二个数据帧包含锚定日期之前的 1000 天。最好的方法是什么？

Answer 1

不知道这是否是最好的方法，但应该有效

创建示例数据帧:

>>> dates = [pd.datetime(2012, 5, 4), pd.datetime(2012, 5, 5), pd.datetime(2012, 5, 6), pd.datetime(2012, 5, 1), pd.datetime(2012, 5, 2), pd.datetime(2012, 5, 3)]
>>> values = [1, 2, 3, 4, 5, 6]
>>> df = pd.DataFrame(values, dates)
>>> df
>>> df
            0
2012-05-04  1
2012-05-05  2
2012-05-06  3
2012-05-01  4
2012-05-02  5
2012-05-03  6

假设我们想要从 2012 年 5 月 4 日起回溯 2 天:

>>> date_end = pd.datetime(2012, 5, 4)
>>> date_start = date_end - pd.DateOffset(days=2)
>>> date_start, date_end
(datetime.datetime(2012, 5, 2, 0, 0), datetime.datetime(2012, 5, 4, 0, 0))

现在让我们尝试按 label indexing 获取行:

>>> df.loc[date_start:date_end]
Empty DataFrame
Columns: [0]
Index: []

这是因为我们的索引未排序，所以让我们修复它:

>>> df.sort_index(inplace=True)
>>> df.loc[date_start:date_end]
            0
2012-05-02  5
2012-05-03  6
2012-05-04  1

也可以通过 datetime indexing 获取行:

>>> df[date_start:date_end]
            0
2012-05-02  5
2012-05-03  6
2012-05-04  1

请记住，我仍然不是 Pandas 方面的专家，但我非常喜欢它进行数据分析。

希望有帮助。