Python股票信息抓取(二)

在一的基础上，想着把所有的折线图放在一个图中，然后图的结果如图所示：

不是略丑，是很丑~

依然的单进程，只是将图标结果放在了一张图里

代码如下：

#-*-coding:utf-8 -*-

import urllib

import re

import json

import urllib2

from lxml import etree

import requests

import time

from Queue import Queue

import matplotlib.pyplot as plt

URL = 'http://quote.fx678.com/exchange/WH'

nation_que = Queue()

nation = Queue()

high = Queue()

start_time = Queue()

Chart = []

def download(url, headers, num_try=2):

    while num_try >0:

        num_try -= 1

        try:

            content = requests.get(url, headers=headers)

            return content.text

        except urllib2.URLError as e:

            print 'Download error', e.reason

    return None

def sub_sort(array,array1,low,high):

    key = array[low]

    key1 = array1[low]

    while low < high:

        while low < high and array[high] >= key:

            high -= 1

        while low < high and array[high] < key:

            array[low] = array[high]

            array1[low] = array1[high]

            low += 1

            array[high] = array[low]

            array1[high] = array1[low]

    array[low] = key

    array1[low] = key1

    return low

def quick_sort(array,array1,low,high):

     if low < high:

        key_index = sub_sort(array,array1,low,high)

        quick_sort(array,array1,low,key_index)

        quick_sort(array,array1,key_index+1,high)

def get_type_url():

    headers = {

        'User_agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36',

        'Referer': 'http://quote.fx678.com/exchange/WH',

        'Cookie': 'io=-voMclEjiizK9nWKALqB; UM_distinctid=15f5938ddc72db-089cf9ba58d9e5-31657c00-fa000-15f5938ddc8b24; Hm_lvt_d25bd1db5bca2537d34deae7edca67d3=1509030420; Hm_lpvt_d25bd1db5bca2537d34deae7edca67d3=1509031023',

        'Accept-Language': 'zh-CN,zh;q=0.8',

        'Accept-Encoding': 'gzip, deflate',

        'Accept': '*/*'

    }

    content = download(URL,headers)

    html = etree.HTML(content)

    result = html.xpath('//a[@class="mar_name"]/@href')

    for each in result:

        print each

        st = each.split('/')

        nation_que.put(st[len(st)-1])

    get_precent()

def get_precent():

    while not nation_que.empty():

        ss = nation_que.get(False)

        print ss

        url = 'http://api.q.fx678.com/history.php?symbol=' + ss +'&limit=288&resolution=5&codeType=8100&st=0.8274405615006541'

        print url

        headers = {'Accept':'application/json, text/javascript, */*; q=0.01',

                'Accept-Encoding':'gzip, deflate',

                'Accept-Language':'zh-CN,zh;q=0.8',

                'Connection':'keep-alive',

                'Host':'api.q.fx678.com',

                'Origin':'http://quote.fx678.com',

                'Referer':'http://quote.fx678.com/symbol/USD',

                'User-Agent':'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/61.0.3163.100 Safari/537.36'

               }

        num_try = 2

        while num_try >0:

            num_try -= 1

            try:

                content = requests.get(url, headers=headers)

                html = json.loads(content.text)

                st = html['h']

                T_time = html['t']

                if  len(st) > 0 and len(T_time) > 0:

                    draw_pict(T_time,st)

                break

            except urllib2.URLError as e:

                print 'Download error', e.reason

        nation_que.task_done()

List = []

def draw_pict(T_time1,high_rate):

    print T_time1

    print high_rate

    num = 0

    High = T_time1

    Time = high_rate

    High_Rate = []

    num = 0

    T_time = []

    for each,high1 in zip(T_time1,high_rate):

        st = time.localtime(float(each))

        if st.tm_min == 0:

            T_time.append(st.tm_hour)

            High_Rate.append(high1)

        else:

            pass

    if len(T_time) == len(High_Rate):

        quick_sort(T_time,High_Rate,0,len(High_Rate)-1)

        List.append(High_Rate)

def draw_picture():

    colu = len(List)

    num = 1

    for each in List:

        plt.subplot(colu/2 + 1,2,num)

        num+=1

        list = each

        T_time = []

        for i in range(len(list)):

            T_time.append(i)

        print len(list)

        print len(T_time)

        plt.plot(T_time, list, marker='*')

    plt.show()

    plt.title('Share Message')

if __name__ == '__main__':

    get_type_url()

    draw_picture()

折线图结果如图

ps:1的是将图的结果进行单个显示，每支股票是单独的一张图