forked from jerryxyx/TreasuryFutureTrading
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathutils.py
19 lines (15 loc) · 789 Bytes
/
utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import pandas as pd
def disaggregateInputData(df):
dfPrices = df.iloc[:, 0:6]
dfDurations = df.iloc[:, [0, 6, 7, 8, 9, 10]]
dfOptWeights = df.iloc[:, [0, 11, 12, 13, 14, 15]]
dfRollingStats = df.loc[:, ["DateTime", "RollingAvg", "RollingStd"]]
dfPrices.set_index("DateTime", inplace=True)
dfDurations.set_index("DateTime", inplace=True)
dfOptWeights.set_index("DateTime", inplace=True)
dfRollingStats.set_index("DateTime", inplace=True)
dfPrices.index = pd.to_datetime(dfPrices.index)
dfDurations.index = pd.to_datetime(dfDurations.index)
dfOptWeights.index = pd.to_datetime(dfOptWeights.index)
dfRollingStats.index = pd.to_datetime(dfRollingStats.index)
return (dfPrices, dfDurations, dfOptWeights, dfRollingStats)