根据日期将数据框分成两部分

新手上路,请多包涵

我有这样的 1000 行数据集

 Date,      Cost,         Quantity(in ton),    Source,          Unloading Station
    01/10/2015, 7,            5.416,               XYZ,           ABC

我想根据日期拆分数据。例如,直到日期 20.12.2016 是训练数据,之后是测试数据。

我应该怎么分?是否可以?

原文由 kush 发布,翻译遵循 CC BY-SA 4.0 许可协议

阅读 392
1 个回答

您可以通过将您的列转换为 pandas to_datetime 类型并将其设置为索引来轻松地做到这一点。

 import pandas as pd
df['Date'] = pd.to_datetime(df['Date'])
df = df.set_index(df['Date'])
df = df.sort_index()

一旦你有了这种格式的数据,你就可以简单地使用日期作为索引来创建分区,如下所示:

 # create train test partition
train = df['2015-01-10':'2016-12-20']
test  = df['2016-12-21':]
print('Train Dataset:',train.shape)
print('Test Dataset:',test.shape)

原文由 Sayali Sonawane 发布,翻译遵循 CC BY-SA 4.0 许可协议

撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题