diff --git a/main.py b/main.py index 3c32127..77fb95d 100644 --- a/main.py +++ b/main.py @@ -73,10 +73,15 @@ if __name__ == '__main__': print("index:") print(data.index) + data = data.drop('Volume', axis=1, level=0) # drop Volume data print(data) + print(data.keys()) + print(data.iloc[0]) print(get_day_x_delta_data(data, 1, True)) - print(get_day_x_delta_data(data, 1)) + d1 = get_day_x_delta_data(data, 1) + print(d1.corr()) + exit() print(data.isnull().sum() / data.shape[0] * 100)