#下載台股台積電:2330
df = get_tw_stock_data(start_year = 2021, 
                       start_month = 9, 
                       end_year = 2021, 
                       end_month = 10, 
                       stock_code = 2330)
df

# 設定子圖
fig = make_subplots(rows = 2, 
                    cols = 1, 
                    shared_xaxes = True, 
                    vertical_spacing = 0.05,  
                    row_width=[0.2, 0.7])

# 畫Ｋ線圖
fig.add_trace(go.Candlestick(x = df['日期'], 
                             open = df['開盤價'], 
                             high = df['最高價'],
                             low = df['最低價'], 
                             close = df['收盤價'], 
                             increasing_line_color = 'red', 
                             decreasing_line_color = 'green',
                             name = 'K線圖'), 
              row = 1, 
              col = 1)
 
# 畫成交量長條圖
fig.add_trace(go.Bar(x = df['日期'], 
                     y = df['成交股數'], 
                     showlegend = False,
                     name = '成交股數'), 
              row = 2, 
              col = 1)
    

# 設x軸標題
fig.update_xaxes(rangebreaks = [{ 'pattern': 'day of week', 'bounds': [6, 1]}])
fig.update_xaxes(title_text = "日期", row = 2, col = 1)

# 設y軸標題
fig.update_yaxes(title_text = "股價", row = 1, col = 1)


# 設圖標及圖長寬
fig.update_layout(
    title_text = "2021/09 - 2021/10 台積電(2330) K線圖",
    width = 800,
    height = 400
)

fig.update(layout_xaxis_rangeslider_visible = False)
fig.show()

df["成交量增減"] = df['成交股數'].diff()
df

df["成交量顏色"] = np.where(df["成交量增減"] >= 0, 'red', 'green')
df

df

# 設定子圖
fig = make_subplots(rows = 2, 
                    cols = 1, 
                    shared_xaxes = True, 
                    vertical_spacing = 0.05,  
                    row_width=[0.2, 0.7])

# 畫Ｋ線圖
fig.add_trace(go.Candlestick(x = df['日期'], 
                             open = df['開盤價'], 
                             high = df['最高價'],
                             low = df['最低價'], 
                             close = df['收盤價'], 
                             increasing_line_color = 'red', 
                             decreasing_line_color = 'green',
                             name = 'K線圖'), 
              row = 1, 
              col = 1)
 
# 畫成交量長條圖
fig.add_trace(go.Bar(x = df['日期'], 
                     y = df['成交股數'], 
                     showlegend = False,
                     name = '成交股數',
                     marker_color = df['成交量顏色']), 
              row = 2, 
              col = 1)
    

# 設x軸標題
fig.update_xaxes(rangebreaks = [{ 'pattern': 'day of week', 'bounds': [6, 1]}])
fig.update_xaxes(title_text = "日期", row = 2, col = 1)

# 設y軸標題
fig.update_yaxes(title_text = "股價", row = 1, col = 1)


# 設圖標及圖長寬
fig.update_layout(
    title_text = "2021/09 - 2021/10 台積電(2330) K線圖",
    width = 800,
    height = 400
)

fig.update(layout_xaxis_rangeslider_visible = False)
fig.show()

twse = get_stock_market_price(start_year = 2021, 
                              start_month = 9, 
                              end_year = 2021, 
                              end_month = 10)
twse

twse_vol = get_stock_market_volumn(start_year = 2021, 
                                   start_month = 9, 
                                   end_year = 2021, 
                                   end_month = 10)
twse_vol

# 設定子圖
fig = make_subplots(rows = 2, 
                    cols = 1, 
                    shared_xaxes = True, 
                    vertical_spacing = 0.05,  
                    row_width=[0.2, 0.7])

# 畫Ｋ線圖
fig.add_trace(go.Candlestick(x = twse['日期'], 
                             open = twse['開盤價'], 
                             high = twse['最高價'],
                             low = twse['最低價'], 
                             close = twse['收盤價'], 
                             increasing_line_color = 'red', 
                             decreasing_line_color = 'green',
                             name = 'K線圖'), 
              row = 1, 
              col = 1)

# 畫成交量長條圖
fig.add_trace(go.Bar(x = twse_vol['日期'], 
                     y = twse_vol['成交金額'], 
                     showlegend = False,
                     name = '成交金額'), 
              row = 2, 
              col = 1)
    

# 設x軸標題
fig.update_xaxes(rangebreaks = [{ 'pattern': 'day of week', 'bounds': [6, 1]}])
fig.update_xaxes(title_text = "日期", row = 2, col = 1)

# 設y軸標題
fig.update_yaxes(title_text = "指數", row = 1, col = 1)


# 設圖標及圖長寬
fig.update_layout(
    title_text = "2021/09 - 2021/10 大盤K線圖",
    width = 800,
    height = 400
)

fig.update(layout_xaxis_rangeslider_visible = False)
fig.show()

twse

twse['成交金額'] = twse_vol['成交金額']
twse

twse["成交金額增減"] = twse['成交金額'].diff()
twse

twse["成交金額顏色"] = np.where(twse["成交金額增減"] >= 0, 'red', 'green')
twse

twse

# 設定子圖
fig = make_subplots(rows = 2, 
                    cols = 1, 
                    shared_xaxes = True, 
                    vertical_spacing = 0.05,  
                    row_width=[0.2, 0.7])

# 畫Ｋ線圖
fig.add_trace(go.Candlestick(x = twse['日期'], 
                             open = twse['開盤價'], 
                             high = twse['最高價'],
                             low = twse['最低價'], 
                             close = twse['收盤價'], 
                             increasing_line_color = 'red', 
                             decreasing_line_color = 'green',
                             name = 'K線圖'), 
              row = 1, 
              col = 1)
 
# 畫成交量長條圖
fig.add_trace(go.Bar(x = twse['日期'], 
                     y = twse['成交金額'], 
                     showlegend = False,
                     marker_color = twse['成交金額顏色'],
                     name = '成交金額'), 
              row = 2, 
              col = 1)
    

# 設x軸標題
fig.update_xaxes(rangebreaks = [{ 'pattern': 'day of week', 'bounds': [6, 1]}])
fig.update_xaxes(title_text = "日期", row = 2, col = 1)

# 設y軸標題
fig.update_yaxes(title_text = "指數", row = 1, col = 1)


# 設圖標及圖長寬
fig.update_layout(
    title_text = "2021/09 - 2021/10 大盤K線圖",
    width = 800,
    height = 400
)

fig.update(layout_xaxis_rangeslider_visible = False)
fig.show()