上一个帖子,已经用python 把我们需要的股票信息从东方财富网给下载到本地了,这一次讲的是怎么把这些数据入库
准备知识: pandas库,pymysql库,mysql数据库软件
我采用的是mysql数据库,之前的帖子 https://www.cnblogs.com/effortscodepanda/p/7631722.html 讲了怎么在windows下安装mysql,如果忘记安装过的话,打开服务看有没有mysql即可,打开方式 wiindows + R 键,然后输入 services.msc即可
安装过的话,登陆即会出现以下界面:
然后安装三方库:pymysql
pip install pymysql ----python 在有 pip工具的情况下,下载三方库和其他功能模块非常方便,不得不吐槽一下c++,哈哈
然后安装pandas,方法同样的:
pip install pandas
不过报错了,遇到了如下的错误:No matching distribution found for pandas
pip install numpy --失败
python -m pip install --upgrade pip --失败
pip install --upgrade pip –vvv --查找错误原因 -失败
python -m pip install -U --force-reinstall pip --失败
pip install anaconda3
pip install -i https://pypi.douban.com/simple pandas --成功!!!!
无力吐槽,windows安装一下各种失败
最终也不是完全成功,不过验证,pandas库可用,有时候借鉴别人开源代码也是很快的:
#导入需要使用到的模块 import urllib import urllib.request import re import os import pandas as pd import pymysql ##########################将股票数据存入数据库########################### filepath = 'D:\\data\\python\\stock\\'#定义数据文件保存路径 #数据库名称和密码 name = 'root' password = 'Test_123' #替换为自己的账户名和密码 #建立本地数据库连接(需要先开启数据库服务) db = pymysql.connect('localhost', name, password, charset='utf8') cursor = db.cursor() #创建数据库stockDataBase sqlSentence1 = "create database stockDataBase" cursor.execute(sqlSentence1)#选择使用当前数据库 sqlSentence2 = "use stockDataBase;" cursor.execute(sqlSentence2) #获取本地文件列表 fileList = os.listdir(filepath) #依次对每个数据文件进行存储 for fileName in fileList: data = pd.read_csv(filepath+fileName, encoding="gbk") #创建数据表,如果数据表已经存在,会跳过继续执行下面的步骤print('创建数据表stock_%s'% fileName[0:6]) sqlSentence3 = "create table stock_%s" % fileName[0:6] + "(日期 date, 股票代码 VARCHAR(10), 名称 VARCHAR(10),\ 收盘价 float, 最高价 float, 最低价 float, 开盘价 float, 前收盘 float, 涨跌额 float, \ 涨跌幅 float, 换手率 float, 成交量 bigint, 成交金额 bigint, 总市值 bigint, 流通市值 bigint)" cursor.execute(sqlSentence3) #迭代读取表中每行数据,依次存储(整表存储还没尝试过) print('正在存储stock_%s'% fileName[0:6]) length = len(data) for i in range(0, length): record = tuple(data.loc[i]) #插入数据语句 try: sqlSentence4 = "insert into stock_%s" % fileName[0:6] + "(日期, 股票代码, 名称, 收盘价, 最高价, 最低价, 开盘价, 前收盘, 涨跌额, 涨跌幅, 换手率, \ 成交量, 成交金额, 总市值, 流通市值) values ('%s',%s','%s',%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)" % record #获取的表中数据很乱,包含缺失值、Nnone、none等,插入数据库需要处理成空值 sqlSentence4 = sqlSentence4.replace('nan','null').replace('None','null').replace('none','null') cursor.execute(sqlSentence4) except: #如果以上插入过程出错,跳过这条数据记录,继续往下进行 break #关闭游标,提交,关闭数据库连接 cursor.close() db.commit() db.close() ###########################查询刚才操作的成果################################## #重新建立数据库连接 db = pymysql.connect('localhost', name, password, 'stockDataBase') cursor = db.cursor() #查询数据库并打印内容 cursor.execute('select * from stock_600000') results = cursor.fetchall() for row in results: print(row) #关闭 cursor.close() db.commit() db.close()
最终跑完之后,查询为空,一度以为是识别不了中文,但是查数据可见:
表是建成功的,因此就是:
#迭代读取表中每行数据,依次存储(整表存储还没尝试过) print('正在存储stock_%s'% fileName[0:6]) length = len(data) for i in range(0, length): record = tuple(data.loc[i]) #插入数据语句 try: sqlSentence4 = "insert into stock_%s" % fileName[0:6] + "(日期, 股票代码, 名称, 收盘价, 最高价, 最低价, 开盘价, 前收盘, 涨跌额, 涨跌幅, 换手率, \ 成交量, 成交金额, 总市值, 流通市值) values ('%s',%s','%s',%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)" % record #获取的表中数据很乱,包含缺失值、Nnone、none等,插入数据库需要处理成空值 sqlSentence4 = sqlSentence4.replace('nan','null').replace('None','null').replace('none','null') cursor.execute(sqlSentence4) except: #如果以上插入过程出错,跳过这条数据记录,继续往下进行 break
这一块代码出错,下一章节讲的就是调试代码了
写代码的小熊猫~:)