drewfustin

{"nbformat_minor": 0, "cells": [{"execution_count": 68, "cell_type": "code", "source": "import pandas as pd\nimport numpy as np\nfrom science.loaders.forecasting.single_well_prod import model_battle_royale\n\nfc_field = 'monthly_well_oil_prod'\nmodels = ['hw', 'aa', 'mc']\n\ndf = pd.DataFrame({\n        '{}'.format(fc_field): [7, 10, 12, 12, 14, np.nan],\n        '{}_{}_fc'.format(fc_field, models[0]): [np.nan, 10, 11, 12, 13, 14],\n        '{}_{}_lb'.format(fc_field, models[0]): [np.nan, 9, 10, 10, 11, 12],\n        '{}_{}_ub'.format(fc_field, models[0]): [np.nan, 11, 12, 14, 16, 17],\n        '{}_{}_fc'.format(fc_field, models[1]): [np.nan, 12, 9, 5, 4, 4],\n        '{}_{}_lb'.format(fc_field, models[1]): [np.nan, 9, 7, 3, 1, 0],\n        '{}_{}_ub'.format(fc_field, models[1]): [np.nan, 16, 11, 8, 6, 6],\n        '{}_{}_fc'.format(fc_field, models[2]): [np.nan, 10, 11, 12, 10, 11],\n        '{}_{}_lb'.format(fc_field, models[2]): [np.nan, 7, 8, 9, 6, 6],\n        '{}_{}_ub'.format(fc_field, models[2]): [np.