Python抓取股票数据

Halcom

Python抓取股票数据

1. # -*- coding: utf-8 -*-
   
2. """
   
3. Spyder Editor
   
4. This is a temporary script file.
   
5. """
   
6. from urllib.parse import urlencode
   
7. import pandas as pd
   
8. import requests
   
9. def gen_eastmoney_code(rawcode: str) -> str:
   
10.     '''
    
11.     生成东方财富专用的secid
    
12. 
    
13.     Parameters
    
14.     ----------
    
15.     rawcode ： 6 位股票代码
    
16.     Parameters
    
17.     ----------
    
18.     str : 按东方财富格式生成的字符串
    
19.     '''
    
20.     if rawcode[0] != '6':
    
21.         return f'0.{rawcode}'
    
22.     return f'1.{rawcode}'
    
23. def get_k_history(code: str, beg: str = '16000101', end: str = '20500101', klt: int = 1, fqt: int = 1) -> pd.DataFrame:
    
24.     '''
    
25.     功能获取k线数据
    
26.     Parameters
    
27.     ----------
    
28.     code : 6 位股票代码
    
29.     beg: 开始日期 例如 20200101
    
30.     end: 结束日期 例如 20200201
    
31.     klt: k线间距 默认为 101 即日k
    
32.         klt:1 1 分钟
    
33.         klt:5 5 分钟
    
34.         klt:101 日
    
35.         klt:102 周
    
36.     fqt: 复权方式
    
37.         不复权 : 0
    
38.         前复权 : 1
    
39.             后复权 : 2
    
40.     Return
    
41.     ------
    
42.     DateFrame : 包含股票k线数据
    
43.     '''
    
44.     EastmoneyKlines = {
    
45.         'f51': '时间',
    
46.         'f52': '开盘',
    
47.         'f53': '收盘',
    
48.         'f54': '最高',
    
49.         'f55': '最低',
    
50.         'f56': '成交量',
    
51.         'f57': '成交额',
    
52.         'f58': '振幅',
    
53.         'f59': '涨跌幅',
    
54.         'f60': '涨跌额',
    
55.         'f61': '换手率',
    
56.     }
    
57.     EastmoneyHeaders = {
    
58.         'User-Agent': 'Mozilla/5.0 (Windows NT 6.3; WOW64; Trident/7.0; Touch; rv:11.0) like Gecko',
    
59.         'Accept': '*/*',
    
60.         'Accept-Language': 'zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2',
    
61.         'Referer': 'http://quote.eastmoney.com/center/gridlist.html',
    
62.     }
    
63.     fields = list(EastmoneyKlines.keys())
    
64.     columns = list(EastmoneyKlines.values())
    
65.     fields2 = ",".join(fields)
    
66.     secid = gen_eastmoney_code(code)
    
67.     params = (
    
68.         ('fields1', 'f1,f2,f3,f4,f5,f6,f7,f8,f9,f10,f11,f12,f13'),
    
69.         ('fields2', fields2),
    
70.         ('beg', beg),
    
71.         ('end', end),
    
72.         ('rtntype', '6'),
    
73.         ('secid', secid),
    
74.         ('klt', f'{klt}'),
    
75.         ('fqt', f'{fqt}'),
    
76.     )
    
77.     base_url = 'https://push2his.eastmoney.com/api/qt/stock/kline/get'
    
78.     url = base_url+'?'+urlencode(params)
    
79.     json_response = requests.get(
    
80.         url, headers=EastmoneyHeaders).json()
    
81.     data = json_response['data']
    
82.     # code = data['code']
    
83.     # 股票名称
    
84.     # name = data['name']
    
85.     klines = data['klines']
    
86.     rows = []
    
87.     for _kline in klines:
    
88.         kline = _kline.split(',')
    
89.         rows.append(kline)
    
90.     df = pd.DataFrame(rows, columns=columns)
    
91.     return df
    
92. if __name__ == "__main__":
    
93.     # 股票代码
    
94.     code = '002503'
    
95.     beg= '16000101'
    
96.     end= '20500101'
    
97.     klt = 101
    
98.     fqt = 1
    
99.     # 根据股票代码、开始日期、结束日期获取指定股票代码指定日期区间的k线数据
    
100.     df = get_k_history(code, beg, end, klt, fqt)
     
101.     # 保存k线数据到表格里面
     
102.     spath = r'D:\994-WiTiAi\WITIAI_InvestmentIncome\Python'
     
103.     # df.to_csv(spath +'//'+ f'{code}.csv', encoding='utf-8-sig', index=None)
     
104.     df.to_csv(spath + '//' + f'{code}.csv', encoding='utf-8-sig', index=None)
     
105.     print(f'股票代码：{code} 的 k线数据已保存到代码目录下的 {code}.csv 文件中')
     
106.    

复制代码

参考：【1】如何用 python 获取实时的股票数据？