我有一个df,它看起来是这样的:
df.head()
Close
Date
2011-12-31 4.472624
2012-01-01 4.680778
2012-01-02 5.000000
2012-01-03 5.145917
2012-01-04 5.228729
我想把这些数据分成两部分:
# Split data
df_train = df['2017':'2020-04'].iloc[:, 0:1].values #an array of Close values
df_test = df['2020-04':].iloc[:, 0:1].values #an array of Close values
。。。引起:
密钥错误:“2017”
年包含在数据帧中。
如有任何帮助,我们将不胜感激。
您需要将索引转换为日期时间
df.index=pd.to_datetime(df.index)
import pandas as pd
df.index = df.index.astype('datetime64[ns]')
df_train = df[(df.index > '2017-01-01') & (df.index < '2020-04-01')]
df_test = df[df.index > '2020-04-01']