iT邦幫忙

2019 iT 邦幫忙鐵人賽

DAY 7
0
AI & Data

量化投資與機器學習研究系列 第 7

07.爬股票代號、產業別

  • 分享至 

  • xImage
  •  

爬取所有股票代號

之後的所有資料都會常常根據這張表來查詢爬蟲

from IPython.core.display import HTML
from pyquery import PyQuery as pq
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'

columns = ['dtype', 'code', 'name', '國際證券辨識號碼', '上市日', '市場別', '產業別', 'CFI']

items = []
for url in [TWSE_URL, TPEX_URL]:
    response_dom = pq(url)
    for tr in response_dom('.h4 tr:eq(0)').next_all().items():
        if tr('b'):
            dtype = tr.text()
        else:
            row = [td.text() for td in tr('td').items()]
            code, name = row[0].split('\u3000')
            items.append(dict(zip(columns, [dtype, code, name, *row[1: -1]])))

data = pd.DataFrame(items)

HTML(data.head().to_html())
<table border="1" class="dataframe">
  <thead>
    <tr style="text-align: right;">
      <th></th>
      <th>CFI</th>
      <th>code</th>
      <th>dtype</th>
      <th>name</th>
      <th>上市日</th>
      <th>國際證券辨識號碼</th>
      <th>市場別</th>
      <th>產業別</th>
    </tr>
  </thead>
  <tbody>
    <tr>
      <th>0</th>
      <td>ESVUFR</td>
      <td>1101</td>
      <td>股票</td>
      <td>台泥</td>
      <td>1962/02/09</td>
      <td>TW0001101004</td>
      <td>上市</td>
      <td>水泥工業</td>
    </tr>
    <tr>
      <th>1</th>
      <td>ESVUFR</td>
      <td>1102</td>
      <td>股票</td>
      <td>亞泥</td>
      <td>1962/06/08</td>
      <td>TW0001102002</td>
      <td>上市</td>
      <td>水泥工業</td>
    </tr>
    <tr>
      <th>2</th>
      <td>ESVUFR</td>
      <td>1103</td>
      <td>股票</td>
      <td>嘉泥</td>
      <td>1969/11/14</td>
      <td>TW0001103000</td>
      <td>上市</td>
      <td>水泥工業</td>
    </tr>
    <tr>
      <th>3</th>
      <td>ESVUFR</td>
      <td>1104</td>
      <td>股票</td>
      <td>環泥</td>
      <td>1971/02/01</td>
      <td>TW0001104008</td>
      <td>上市</td>
      <td>水泥工業</td>
    </tr>
    <tr>
      <th>4</th>
      <td>ESVUFR</td>
      <td>1108</td>
      <td>股票</td>
      <td>幸福</td>
      <td>1990/06/06</td>
      <td>TW0001108009</td>
      <td>上市</td>
      <td>水泥工業</td>
    </tr>
  </tbody>
</table>
data['dtype'].value_counts()
上市認購(售)權證       15763
上櫃認購(售)權證        5008
股票               1684
ETF               142
臺灣存託憑證(TDR)        17
受益證券-資產基礎證券        10
特別股                 9
受益證券-不動產投資信託        6
臺灣存託憑證              1
Name: dtype, dtype: int64

確定可以爬後改用scrapy重寫,把資料寫入到DB
因為股票會更新,定期刪掉重爬更新這張表

import scrapy

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'

columns = ['dtype', 'code', 'name', '國際證券辨識號碼', '上市日', '市場別', '產業別', 'CFI']


class StockCodeSpider(scrapy.Spider):
    name = 'stock_code'
    start_urls = [TWSE_URL, TPEX_URL]

    custom_settings = {
        'DOWNLOAD_DELAY': 1,
        'CONCURRENT_REQUESTS': 1,
        'MONGODB_COLLECTION': name,
        'MONGODB_ITEM_CACHE': 1000,
        'MONGODB_DROP': True
    }

    def parse(self, response):
        for tr in response.dom('.h4 tr:eq(0)').next_all().items():
            if tr('b'):
                dtype = tr.text()
            else:
                row = [td.text() for td in tr('td').items()]
                code, name = row[0].split('\u3000')
                yield dict(zip(columns, [dtype, code, name, *row[1: -1]]))

每隔一段時間就會有新上市櫃公司或下市櫃,而且這張表未來會常常使用,如果出狀況影響後面流程很大,所以設定成每個月爬一次並把這張表一開始先全部刪除再重新爬3次。

    start_urls = [TWSE_URL, TPEX_URL]*3

    custom_settings = {
        'DOWNLOAD_DELAY': 30,
        'CONCURRENT_REQUESTS': 1,
        'MONGODB_COLLECTION': name,
        'MONGODB_ITEM_CACHE': 1000,
        'MONGODB_UNIQ_KEY': [("code", 1)],
        'MONGODB_DROP': True
    }

上一篇
06.爬蟲工具pyquery用法
下一篇
08.爬股市每日價、量
系列文
量化投資與機器學習研究30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中
0
wilson652033
iT邦新手 5 級 ‧ 2018-11-08 14:20:05

請問大大
1.我安裝的scrapy目錄中有一個settings.py
請問使用了custom_settings 是不是就不用管 settings.py了
2.沒看到大大的程式中有寫入MONGODB的部分---是省略了嗎
謝謝

kuankuan iT邦新手 5 級 ‧ 2018-11-08 20:52:24 檢舉

1.會蓋過去,但沒有更改的設定還是會照settings.py
2.因為不是教scrapy所以沒寫,這個是自己寫的scrapy擴充,建議了解股市資料要去哪裡爬,清理要注意的地方,就可以用自己喜歡的方式爬就好。
網路上很多免費的python爬蟲教學
scrapy的教學
scrapy的文件

0
wilson652033
iT邦新手 5 級 ‧ 2018-11-08 15:18:02

請問大大
1.我執行上述程式碼會報錯
for tr in response.dom('.h4 tr:eq(0)').next_all().items():
AttributeError: 'HtmlResponse' object has no attribute 'dom'
請問我少裝了什麼嗎

kuankuan iT邦新手 5 級 ‧ 2018-11-08 20:48:21 檢舉

把這個response.dom
改成這個就可以跑了
from pyquery import PyQuery
PyQuery(response.text)

0
wilson652033
iT邦新手 5 級 ‧ 2018-11-16 19:09:00

請問kuankuan大大
是如何知道code, name中間的分隔是\u3000
我只看到空格
code, name = row[0].split('\u3000')
謝謝

kuankuan iT邦新手 5 級 ‧ 2018-11-20 09:29:51 檢舉

你直接用 pyquery選出來就看的到了
from pyquery import PyQuery as pq
pq('http://isin.twse.com.tw/isin/C_public.jsp?strMode=4')('.h4 tr:eq(5)').text()

這個結果是
'700004\u3000中美晶群益84購02 TW18Z7000045 2018/10/01 上櫃 RWSCCA'

我要留言

立即登入留言