# Function to predict the price of 1 day
def predict_price_AR(train_data):
   # Define model
   model = ARIMA(train_data, order=(1, 1, 0))

   # Fit the model
   model_fit = model.fit()

   # Make forecast
   forecast_results = model_fit.forecast()

   return forecast_results