turtle/visualize_data.py at master · kanatm287/turtle · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import plotly.graph_objects as go
import turtle_data
import hourly_data
import pre_process
import daily_data


def algo_vs_benchmark(data_frame, algo, benchmark):

    fig = go.Figure()

    fig.add_trace(go.Scatter(x=data_frame.index,
                             y=data_frame[algo],
                             name="algo",
                             line_color="red",
                             opacity=0.8))

    fig.add_trace(go.Scatter(x=data_frame.index,
                             y=data_frame[benchmark],
                             name="benchmark",
                             line_color="blue",
                             opacity=0.8))

    fig.update_layout(title_text="Chart")
    fig.show()


def line_chart(data_frame, columns):

    fig = go.Figure()

    for column in columns:

        fig.add_trace(go.Scatter(x=data_frame.index,
                                 y=data_frame[column["column"]],
                                 name=column["column"],
                                 line_color=column["color"],
                                 opacity=0.8))

    fig.update_layout(title_text="Chart")
    fig.show()


def candle_sticks(data_frame):

    fig = go.Figure(data=[go.Candlestick(x=data_frame["date"],
                                         open=data_frame["open"],
                                         high=data_frame["high"],
                                         low=data_frame["low"],
                                         close=data_frame["close"])])

    fig.show()


# hour_data_frame = turtle_data.prepare(hourly_data.generate(pre_process.minute_data("BTCUSD", 55)), 20, 55, 20)
# daily_data_frame = turtle_data.prepare(daily_data.generate(pre_process.minute_data("BTCUSD", 55)), 20, 55, 20)

# candle_sticks(hour_data_frame)

# columns = ["close",
#            "average_true_range",
#            "dollar_volatility",
#            "average_high_entry",
#            "average_low_entry",
#            "average_high_exit",
#            "average_low_exit",
#            "average_true_range"]

# hour_data_frame = hour_data_frame.set_index("date")
# daily_data_frame = daily_data_frame.set_index("date")
#
# line_chart(daily_data_frame.head(1000),
#            [{"color": "red", "column": "close"},
#             {"color": "green", "column": "high_entry"},
#             {"color": "blue", "column": "low_entry"},
#             {"color": "blue", "column": "average_true_range"}])  # ,
#             # {"color": "brown", "column": "average_high_exit"},
#             # {"color": "black", "column": "average_low_exit"}])