用Python將股票資料視覺化 Plotly套件篇

先看看成果怎樣

會使用到的環境

• Python：3.7.10
• requests: 抓取網頁
• pandas：資料處理用

一樣先從爬蟲開始

爬取台股上市櫃股票分類

import requests
import pandas as pd

TWSE_URL = 'http://isin.twse.com.tw/isin/C_public.jsp?strMode=2' # 上市市場
TPEX_URL = 'http://isin.twse.com.tw/isin/C_public.jsp?strMode=4' # 上櫃市場

res = requests.get("https://isin.twse.com.tw/isin/class_main.jsp?owncode=&stockname=&isincode=&market=1&issuetype=1&industry_code=&Page=1&chklike=Y")
df = pd.read_html(res.text)[0]
 # 資料都在第一個表格裡

# 設定column名稱
df.columns = df.iloc[0]
# 刪除第一行
df = df.iloc[1:]

作圖用的DataFrame

# 這邊是使用dict的方式創建一個新的DataFrame
dff = pd.DataFrame({
    'stock_id': info['有價證券代號'],
    'county': 'TW-Stock', # 用來把下面的路徑包起來
    'label': info['label'],
    'market': info['市場別'],
    'category': info['產業別']
})

Sunburst作圖

import plotly.express as px
 

fig = px.sunburst(
                  dff, 
                  path=['county', 'market', 'category','label'], 
                  values='成交金額', 
                  color='當天漲跌幅', 
                  color_continuous_scale=[(0, "green"), (0.25, 'green'), (0.5, "#f7f7f7"), (0.75, "red"), (1, "red")], 
                  color_continuous_midpoint=0, 
                  maxdepth=2,
                  range_color=[-10.0, 10.0],
                  width=1350,
                  height=900,
                  title='2022/6/17台股成交圖'
)
fig.update_traces(
textinfo="label+percent entry", selector=dict(type='sunburst'), textfont=dict(size=18)
)

fig.update_traces(textfont_size=14)

textinfo="label+percent entry"

那怎麼嵌入wordpress呢?

生成div區塊的html

import plotly as plt
plt.offline.plot(fig, include_plotlyjs=False, output_type='div')

或乾脆弄個網頁崁進去吧

<iframe scrolling="yes" style="zoom:50%" src="https://dow1226.github.io/stock-plotly-test/" height="1280" width="960"> </iframe>

題外話