Как разделить фрейм данных

Question

У меня есть 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

... что вызывает:

KeyError: '2017'

Год включен в фрейм данных.

Любая помощь будет принята с благодарностью.

Answer 1

Вам нужно преобразовать ваш индекс в datetime

df.index=pd.to_datetime(df.index)

Answer 2

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']