以網(wǎng)易股票的股價(jià)為輸入服猪，嘗試使用Altair繪制市值圖：
#!/usr/bin/env python3
import os
import json
import requests
import altair as alt
stocks_file = 'stock.json'
def fetch_stocks():
url = 'http://quotes.money.163.com/hs/service/diyrank.php'
fields = [
'NO',
'SYMBOL',
'NAME',
'PRICE',
'PERCENT',
'UPDOWN',
'FIVE_MINUTE',
'OPEN',
'YESTCLOSE',
'HIGH',
'LOW',
'VOLUME',
'TURNOVER',
'HS',
'LB',
'WB',
'ZF',
'PE',
'MCAP',
'TCAP',
'MFSUM',
'MFRATIO.MFRATIO2',
'MFRATIO.MFRATIO10',
'SNAME',
'CODE',
'ANNOUNMT',
'UVSNEWS'
]
payload = {
'query': 'STYPE:EQA',
'fields': ','.join(fields),
'count': 0
}
response = requests.get(url, params=payload)
if response.status_code != requests.codes.ok:
print(response.reason)
return False
else:
return response.json()
def is_stored():
return os.path.exists(stocks_file) and os.path.isfile(stocks_file)
def store_stocks():
stocks = fetch_stocks()
if not stocks:
return
with open(stocks_file, 'w') as f:
json.dump(stocks, f, ensure_ascii=False, indent=2)
def read_stocks():
if not is_stored():
store_stocks()
with open(stocks_file) as f:
return json.load(f)['list']
def plot():
stocks = alt.Data(values=sorted(read_stocks(),
key=lambda k: -k['TCAP'])[:30])
chart = alt.Chart(stocks).mark_bar().encode(
x=alt.X('TCAP:Q'),
y=alt.Y('NAME:N', sort=alt.SortField(field='TCAP',
order='descending', op='sum'),
scale=alt.Scale(bandSize=13.0,)),
)
chart.serve('0.0.0.0', 8888)
if __name__ == '__main__':
plot()
最終生成市值Top30圖片如下：
市值Top30
最終生成的是靜態(tài)圖巷送，無法互動(dòng)醉拓，還是不如D3.js生成的圖標(biāo)漂亮圈盔。
