011:获取上证50的所有股票代码,并下载各个股K线数到excel表中

news2024/9/22 3:46:18

我们结合《获取上证50的所有股票代码》,《根据股票代码和起始日期获取K线数据到excel表》两文中的脚本,搞出新的脚本:
 

import tkinter as tk
from tkinter import messagebox
from tkcalendar import Calendar
import pandas as pd
import requests
from urllib.parse import urlencode
import requests
from bs4 import BeautifulSoup
import os
import shutil
 
 
def gen_secid(rawcode: str) -> str:
    '''
    生成东方财富专用的secid
    Parameters
    ----------
    rawcode : 6 位股票代码
    Return
    ------
    str: 指定格式的字符串
    '''
    # 沪市指数
    if rawcode[:3] == '000':
        return f'1.{rawcode}'
    # 深证指数
    if rawcode[:3] == '399':
        return f'0.{rawcode}'
    # 沪市股票
    if rawcode[0] != '6':
        return f'0.{rawcode}'
    # 深市股票
    return f'1.{rawcode}'
 
 
def get_k_history(code: str, beg: str, end: str, klt: int = 101, fqt: int = 1) -> pd.DataFrame:
    '''
    功能获取k线数据
    -
    参数
        code : 6 位股票代码
        beg: 开始日期 例如 20200101
        end: 结束日期 例如 20200201
        klt: k线间距 默认为 101 即日k
            klt:1 1 分钟
            klt:5 5 分钟
            klt:101 日
            klt:102 周
        fqt: 复权方式
            不复权 : 0
            前复权 : 1
            后复权 : 2 
    '''
    EastmoneyKlines = {
        'f51': '日期',
        'f52': '开盘',
        'f53': '收盘',
        'f54': '最高',
        'f55': '最低',
        'f56': '成交量',
        'f57': '成交额',
        'f58': '振幅',
        'f59': '涨跌幅',
        'f60': '涨跌额',
        'f61': '换手率',
 
 
    }
    EastmoneyHeaders = {
  
        'User-Agent': 'Mozilla/5.0 (Windows NT 6.3; WOW64; Trident/7.0; Touch; rv:11.0) like Gecko',
        'Accept': '*/*',
        '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',
        'Referer': 'http://quote.eastmoney.com/center/gridlist.html',
    }
    fields = list(EastmoneyKlines.keys())
    columns = list(EastmoneyKlines.values())
    fields2 = ",".join(fields)
    secid = gen_secid(code)
    params = (
        ('fields1', 'f1,f2,f3,f4,f5,f6,f7,f8,f9,f10,f11,f12,f13'),
        ('fields2', fields2),
        ('beg', beg),
        ('end', end),
        ('rtntype', '6'),
        ('secid', secid),
        ('klt', f'{klt}'),
        ('fqt', f'{fqt}'),
    )
    params = dict(params)
    base_url = 'https://push2his.eastmoney.com/api/qt/stock/kline/get'
    url = base_url+'?'+urlencode(params)
    json_response: dict = requests.get(
        url, headers=EastmoneyHeaders).json()
 
    data = json_response.get('data')
    if data is None:
        if secid[0] == '0':
            secid = f'1.{code}'
        else:
            secid = f'0.{code}'
        params['secid'] = secid
        url = base_url+'?'+urlencode(params)
        json_response: dict = requests.get(
            url, headers=EastmoneyHeaders).json()
        data = json_response.get('data')
    if data is None:
        print('股票代码:', code, '可能有误')
        return pd.DataFrame(columns=columns)
 
    klines = data['klines']
 
    rows = []
    for _kline in klines:
 
        kline = _kline.split(',')
        rows.append(kline)
 
    df = pd.DataFrame(rows, columns=columns)
 
    return df
 
def select_start_date():
    def on_date_selected():
        selected_date = cal.selection_get()
        start_date_entry.delete(0, tk.END)
        start_date_entry.insert(0, selected_date.strftime('%Y%m%d'))
        top.destroy()
 
    top = tk.Toplevel(root)
    cal = Calendar(top, selectmode='day')
    cal.pack()
    confirm_button = tk.Button(top, text='确认', command=on_date_selected)
    confirm_button.pack()
 
def select_end_date():
    def on_date_selected():
        selected_date = cal.selection_get()
        end_date_entry.delete(0, tk.END)
        end_date_entry.insert(0, selected_date.strftime('%Y%m%d'))
        top.destroy()
 
    top = tk.Toplevel(root)
    cal = Calendar(top, selectmode='day')
    cal.pack()
    confirm_button = tk.Button(top, text='确认', command=on_date_selected)
    confirm_button.pack()
 
def get_kline_data(code,index):
    # code = stock_code_entry.get()
    start_date = start_date_entry.get()
    end_date = end_date_entry.get()
     # 修改文件保存的位置
    save_path = os.path.join('sz50_all_k_data', f'{code}.csv')

    try:
        df = get_k_history(code, start_date, end_date)
        df.to_csv(save_path, encoding='utf-8-sig', index=None)
        print(index,'提示', f'股票代码:{code} 的 k线数据已保存到代码目录下的 {code}.csv 文件中')
    except:
        print(index,'错误',{code}, '获取K线数据失败')

def get_sz50_all_data():
    url = "https://q.stock.sohu.com/cn/bk_4272.shtml"
    response = requests.get(url)
    soup = BeautifulSoup(response.text, "html.parser")

    # 找到包含class为'e1'的元素
    elements = soup.find_all(class_="e1")

    # 提取数据并剔除非数字的成员
    data_list = [element.text for element in elements if element.text.isdigit()]

    # 打印list最终的成员
    print(data_list)

    #遍历所有成员,并调用get_kline_data

    # 检查并创建目录
    if not os.path.exists('sz50_all_k_data'):
        os.makedirs('sz50_all_k_data')
    else:
        shutil.rmtree('sz50_all_k_data')
        os.makedirs('sz50_all_k_data')

    index=0
    for code_tmp in data_list:
        index += 1
        get_kline_data(code_tmp,index)
    
    print(">>>>>>>>>>>>>>>>完成")

root = tk.Tk()
root.title('上证50所有个股数据获取')
 
# stock_code_label = tk.Label(root, text='股票代码')
# stock_code_label.pack()
# stock_code_entry = tk.Entry(root)
# stock_code_entry.pack()
 
start_date_label = tk.Label(root, text='起始日期')
start_date_label.pack()
start_date_entry = tk.Entry(root)
start_date_entry.pack()
 
select_start_date_button = tk.Button(root, text='选择日期', command=select_start_date)
select_start_date_button.pack()
 
end_date_label = tk.Label(root, text='结束日期')
end_date_label.pack()
end_date_entry = tk.Entry(root)
end_date_entry.pack()
 
select_end_date_button = tk.Button(root, text='选择日期', command=select_end_date)
select_end_date_button.pack()
 
get_data_button = tk.Button(root, text='获取K线数据', command=get_sz50_all_data)
get_data_button.pack()
 
root.mainloop()

效果:

保存到名为sz50_all_k_data的文件夹中:

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1080398.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

阿里8年经验之谈 —— 如何编写有效的接口测试?

阿里妹导读:在所有的开发测试中,接口测试是必不可少的一项。有效且覆盖完整的接口测试,不仅能保障新功能的开发质量,还能让开发在修改功能逻辑的时候有回归的能力,同时也是能优雅地进行重构的前提。编写接口测试要遵守…

web前端面试-- js深拷贝的一些bug,特殊对象属性(RegExp,Date,Error,Symbol,Function)处理,循环引用weekmap处理

本人是一个web前端开发工程师,主要是vue框架,整理了一些面试题,今后也会一直更新,有好题目的同学欢迎评论区分享 ;-) web面试题专栏:点击此处 文章目录 深拷贝和浅拷贝的区别浅拷贝示例深拷贝示例 特殊对象…

LeetCode-145-二叉树的后序遍历

文章目录 递归法非递归法 题目描述&#xff1a; 题目链接&#xff1a;LeetCode-145-二叉树的后序遍历 递归法 解题思路&#xff1a;递归&#xff0c;具体可以参考 LeetCode-144-二叉树的前序遍历 代码实现&#xff1a; class Solution {List<Integer> listnew ArrayList…

数据结构中常见的排序及其代码C语言版本

这里非常抱歉 没有做好动图 等我做好了动态单独做一期博客到时候大家结合动图将本篇博客联系起来看 更容易理解一些喔 常见的数据结构中的排序算法有冒泡排序、选择排序、插入排序、希尔排序、归并排序、快速排序、堆排序等。 冒泡排序&#xff1a;比较相邻的元素&#xff0c;…

京东数据分析软件:京东销售数据在哪里看?

京东数据分析可以帮助企业了解京东平台上的销售数据和用户行为数据&#xff0c;以便更有效地制定营销策略和优化经营。例如&#xff0c;通过分析京东销量数据&#xff0c;店家可以了解自己店铺的销售情况&#xff0c;发现问题并及时调整。这对于店家来说具有重要的作用&#xf…

塔望食观察 | 中国海参产业发展现状及挑战

海参&#xff0c;一个古老的物种&#xff0c;堪称海底活化石&#xff0c;据资料显示&#xff0c;海参在地球上存活超过6亿年&#xff0c;比恐龙还早。海参的药用、食疗和营养滋补价值极高&#xff0c;清朝学者赵学敏编的《本草纲目拾遗》有这样的叙述&#xff1a;“海参性温补&…

SpringBoot通过配置切换注册中心(多注册中心nacos和eureka)

场景&#xff1a; 因项目需要&#xff0c;一个springcloud微服务工程需要同时部署到A,B两个项目使用&#xff0c;但A项目使用Eureka注册中心&#xff0c;B项目使用Nacos注册中心&#xff0c;现在需要通过部署时修改配置来实现多注册中心的切换。 解决思路&#xff1a; 如果同时…

算法通关村第17关【黄金】| 跳跃游戏问题

1. 跳跃游戏 思路&#xff1a;每个nums[i]i代表当前最后覆盖范围&#xff0c;这里和字符串分割那道题一样求覆盖区间&#xff0c;取cur的最大值 class Solution {public boolean canJump(int[] nums) {int len nums.length;if(len 1){return true;}int cur 0;for(int i 0;…

如何建立线上线下相结合的数字化新零售体系?

身处今数字化时代&#xff0c;建立线上线下相结合的数字化新零售体系是企业成功的关键。蚓链数字化营销系统致力于帮助企业实现数字化转型&#xff0c;打通线上线下销售渠道&#xff0c;提升品牌影响力和用户黏性&#xff0c;那么具体是如何建立的&#xff1f; 1. 搭建数字化中…

虹科分享 | 确保冻干工艺开发中精确测量和数据完整性的5步指南

虹科分享 | 确保冻干工艺开发中精确测量和数据完整性的5步指南 介绍 冻干周期的工艺开发在冻干中起着至关重要的作用&#xff0c;因为它可以优化关键工艺参数&#xff0c;以实现理想的产品质量和工艺一致性。优化冻干工艺还可以缩短运行时间&#xff0c;尽早发现关键错误&…

基于threejs的3d可视化工厂示例

源码下载地址在文章末尾&#xff01; 效果演示 总览 内部设备 选中机器展示数据 源码下载地址 https://download.csdn.net/download/qq_43185384/88415706

torch-scatter安装失败解决办法

安装torch-scatter时可以直接下载whl安装&#xff0c;但是会遇到版本不一致的问题&#xff0c;虽然可以安装但是会提示.so文件不存在或者undefined symbol: 的问题&#xff0c;可以在下面的地址中找到跟自己的torch和cuda版本对应的torch-scatter安装文件。 https://pytorch-ge…

我的AI存储实践及思考

前言 我从2020年进入人工智能行业&#xff0c;开始为AI训练提供存储解决方案及技术支持&#xff0c;随着这几年行业以及公司的发展&#xff0c;采用的存储方案经历了几次大的演变 &#xff0c;最开始使用的是分布式并行文件系统Lustre&#xff0c;接着是BeeGFSCeph&#xff0c…

leetcode(力扣) 221. 最大正方形(动态规划)

文章目录 题目描述思路分析完整代码 题目描述 在一个由 ‘0’ 和 ‘1’ 组成的二维矩阵内&#xff0c;找到只包含 ‘1’ 的最大正方形&#xff0c;并返回其面积。 输入&#xff1a;matrix [[“1”,“0”,“1”,“0”,“0”],[“1”,“0”,“1”,“1”,“1”],[“1”,“1”,“1…

SQLAlchemy 使用封装实例

类封装 database.py #! /usr/bin/env python # -*- coding: utf-8 -*-import sys import json import logging from datetime import datetimefrom core.utils import classlock, parse_bool from core.config import (MYSQL_HOST,MYSQL_PORT,MYSQL_USER,MYSQL_PASS,MYSQL_DA…

calc方法和vue中calc不生效踩坑

calc方法 calc()方法是css用来计算的,比如一个场景,上下固定高度,中间自适应,就可以使用这个方法。 预编译less也是可以使用这个方法的 <!DOCTYPE html> <html lang"en"> <head><meta charset"UTF-8"><meta name"viewp…

AR动态贴纸SDK,让创作更加生动有趣

在当今的社交媒体时代&#xff0c;视频已经成为了人们表达自我、分享生活的重要方式。然而&#xff0c;如何让你的视频在众多的信息中脱颖而出&#xff0c;吸引更多的关注和点赞呢&#xff1f;答案可能就在你的手中——美摄AR动态贴纸SDK。 美摄AR动态贴纸SDK是一款专为视频编辑…

设计模式02———建造者模式 c#

首先我们打开一个项目 在这个初始界面我们需要做一些准备工作 建基础通用包 创建一个Plane 重置后 缩放100倍 加一个颜色 更换天空盒&#xff08;个人喜好&#xff09; 任务&#xff1a;使用【UI】点击生成6种车零件组装不同类型车 【建造者模式】 首先资源商店下载车模型 将C…

C#实现五子棋小游戏:简单、有趣的编程项目

目录 引言什么是五子棋游戏规则开发环境准备安装C#开发环境选择合适的集成开发环境(IDE)游戏设计与功能分析游戏界面设计实现棋盘的绘制与操作实现落子功能实现输赢判断说明引言 什么是五子棋 五子棋是一种源于中国的传统棋类游戏,常见于中国、日本、韩国等亚洲国家,是亚洲…

c++视觉处理-----cv::findContours函数和图像进行去噪、平滑、边缘检测和轮廓检测,动态检测图形

cv::findContours cv::findContours 是OpenCV中用于查找图像中对象轮廓的函数。轮廓是对象的边界&#xff0c;通常用于对象检测、分割和形状分析。cv::findContours 函数的基本用法如下&#xff1a; cv::findContours(image, contours, hierarchy, mode, method, offset cv:…