借助Tushare接口调用的股票数据制作简单的股票交易小程序
Posted weixin_46259979
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了借助Tushare接口调用的股票数据制作简单的股票交易小程序相关的知识，希望对你有一定的参考价值。
import tushare as ts
ts.set_token('*************')
pro = ts.pro_api()tushare id:433719
小程序采用pyqt5制作，只是很短时间学习并完成的，为学习过程的制作的比较粗糙的小程序，为初学者水平，注释比较详细，也有点啰嗦，代码新手，学习过程中随手写的，多包涵，仅供参考。
Tushare大数据社区，官网所在地址。
首先，在pycharm中调用tushare函数，set_token中为自己申请的密匙
def queren1():
#显示当天买入价格
#data,shouxufei = start()
data = ui.lineEdit_10.text()
codelist = ui.lineEdit_3.text()
number = ui.lineEdit_5.text()
df = pro.daily(ts_code=codelist, trade_date=data)
close = df.iloc[0,5]
k = pro.stock_basic(ts_code=codelist,
fields='ts_code,name,area,industry,list_date')
name = k.iloc[0, 1]
ui.lineEdit_4.setText(str(close))
ui.lineEdit_11.setText(str(name))以上代码为显示当天的买入价格，pro.daily可调用指定股票在这一天的一些基本数据，此外本函数也可调用一些成交量，涨跌额等一些数据，详细可以查看Tushare大数据社区，其中包含更多的功能，本程序采用的是股票这一天的收盘价的数据。
小程序提供了买入模块和卖出模块，代码如下
def buy():
codelist = ui.lineEdit_3.text()
number = ui.lineEdit_5.text()
buyprice = ui.lineEdit_4.text()
shouxufei = ui.lineEdit_9.text()
endprice =float(buyprice)*float(number)
k = pro.stock_basic(ts_code=codelist, fields='ts_code,name,area,industry,list_date')
name = k.iloc[0,1]
for i in range(0,30):
if ui.tableWidget.item(i,1) == None:
#股票名称
ui.tableWidget.setItem(i,0,QTableWidgetItem(str(name)))
#股票代码
ui.tableWidget.setItem(i,1, QTableWidgetItem(codelist))
#持仓数量
ui.tableWidget.setItem(i,2,QTableWidgetItem(number))
#交易价格
ui.tableWidget.setItem(i,3, QTableWidgetItem(buyprice))
#总份额
ui.tableWidget.setItem(i,7, QTableWidgetItem(str(endprice)))
#手续费
ui.tableWidget.setItem(i,8,QTableWidgetItem(str(float(buyprice)*float(number)*float(shouxufei)*0.01)))
break
else:
if codelist == ui.tableWidget.item(i,1).text():
# 持仓数量
backnumber = ui.tableWidget.item(i,2).text()
allnumber = float(number)+float(backnumber)
ui.tableWidget.setItem(i,2,QTableWidgetItem(str(allnumber)))
# 交易价格
backprice = ui.tableWidget.item(i,3).text()
allprice = (float(backprice)*float(backnumber)+float(number)*float(buyprice))/(float(backnumber)+float(number))
ui.tableWidget.setItem(i,3,QTableWidgetItem(str(allprice)))
# 总份额
backendprice = ui.tableWidget.item(i,7).text()
allendprice = float(backendprice)+float(number)*float(buyprice)
ui.tableWidget.setItem(i,7,QTableWidgetItem(str(allendprice)))
# 手续费
backshouxufei = ui.tableWidget.item(i,8).text()
allshouxufei = float(backshouxufei)+float(buyprice) * float(number) * float(shouxufei) * 0.01
ui.tableWidget.setItem(i, 8, QTableWidgetItem(str(allshouxufei)))
break卖出模块同理，不再展示代码。程序还提供了显示了个人资产的总市值，总资产，以及每次买卖股票所得到的盈亏金额与盈亏率，以及单位净值等基本功能，详细可见下图
此外程序提供了每日打开程序后的更新模块即导入之前输入的数据，可直观看出目前持有的股票与当前每个股票的盈亏金额，代码如下。
def gengxin():
data = ui.lineEdit_10.text()
zongfene = 0
zongshizhi = 0
for i in range(1,30):
if ui.tableWidget.item(i,1) == None:
break
else:
codelist = ui.tableWidget.item(i,1).text()
lastclose = ui.tableWidget.item(i,3).text()
lastnumber = ui.tableWidget.item(i,2).text()
#计算总份额
fene = ui.tableWidget.item(i,7).text()
zongfene =float(zongfene) + float(fene)
zongfene = round(zongfene,2)
#获取股票现价
ts.set_token('184fedac133c785faa3197981f94ef25b8b34047471c081bdb6323c8')
pro = ts.pro_api()
df = pro.daily(ts_code=codelist, trade_date=data)
close = df.iloc[0,5]
#现价
ui.tableWidget.setItem(i, 5, QTableWidgetItem(str(close)))
#计算股票总市值（现价）
shizhi = float(close)*float(lastnumber)
zongshizhi =float(zongshizhi) + float(shizhi)
zongshizhi = round(zongshizhi,2)
#盈亏金额
yingkui = (float(close)-float(lastclose))*float(lastnumber)
yingkui =round(yingkui,3)
ui.tableWidget.setItem(i,4,QTableWidgetItem(str(yingkui)))
#盈亏率
yinglv = 100 * yingkui / (float(lastclose) * float(lastnumber))
yinglv = round(yinglv,2)
ui.tableWidget.setItem(i,6,QTableWidgetItem(str(yinglv)))
ui.lineEdit.setText(str(zongfene))
ui.lineEdit.setReadOnly(True)
ui.lineEdit_15.setText(str(zongshizhi))
ui.lineEdit_15.setReadOnly(True)
#总资产=总市值+卖出盈利
yingli = ui.lineEdit_2.text()
zongzichan = float(zongshizhi)+float(yingli)
ui.lineEdit_14.setText(str(zongzichan))
ui.lineEdit_14.setReadOnly(True)
#单位净值
jingzhi = float(zongshizhi)/float(zongfene)
jingzhi = round(jingzhi,2)
ui.lineEdit_13.setText(str(jingzhi))
ui.lineEdit_13.setReadOnly(True)小程序没有打包，只是用来进行学习，完整代码如下，仅包括了主程序未放置qt desigin代码，需要可自行查看
from a import *
import tushare as ts
import datetime
from PyQt5.QtWidgets import *
from b import *
import xlwt
import xlrd
if __name__ == '__main__':
app = QApplication(sys.argv)
MainWindow = QMainWindow()
ui = Ui_MainWindow()
ui.setupUi(MainWindow)
# 实例化子窗口
child = QDialog()
child_ui = Ui_Dialog()
child_ui.setupUi(child)
ts.set_token('184fedac133c785faa3197981f94ef25b8b34047471c081bdb6323c8')
pro = ts.pro_api()
def start():
data = (datetime.datetime.now() - datetime.timedelta(days=1))
data = data.strftime("%Y%m%d")
ui.lineEdit_10.setText(data)
#ui.lineEdit_10.setReadOnly(True)
#请输入手续费
shouxufei = ui.lineEdit_9.text()
return data,shouxufei
ui.pushButton.clicked.connect(start)
def queren1():
#显示当天买入价格
#data,shouxufei = start()
data = ui.lineEdit_10.text()
codelist = ui.lineEdit_3.text()
number = ui.lineEdit_5.text()
df = pro.daily(ts_code=codelist, trade_date=data)
close = df.iloc[0,5]
k = pro.stock_basic(ts_code=codelist, fields='ts_code,name,area,industry,list_date')
name = k.iloc[0, 1]
ui.lineEdit_4.setText(str(close))
ui.lineEdit_11.setText(str(name))
ui.pushButton_6.clicked.connect(queren1)
def queren2():
#显示当天的卖出价格
#data,shouxufei = start()
data = ui.lineEdit_10.text()
codelist1 = ui.lineEdit_6.text()
number1 = ui.lineEdit_8.text()
codelist1 = str(codelist1)
df = pro.daily(ts_code=codelist1, trade_date=data)
close = df.iloc[0, 5]
k = pro.stock_basic(ts_code=codelist1, fields='ts_code,name,area,industry,list_date')
name = k.iloc[0, 1]
ui.lineEdit_7.setText(str(close))
ui.lineEdit_12.setText(str(name))
ui.pushButton_7.clicked.connect(queren2)
def gengxin():
data = ui.lineEdit_10.text()
zongfene = 0
zongshizhi = 0
for i in range(1,30):
if ui.tableWidget.item(i,1) == None:
break
else:
codelist = ui.tableWidget.item(i,1).text()
lastclose = ui.tableWidget.item(i,3).text()
lastnumber = ui.tableWidget.item(i,2).text()
#计算总份额
fene = ui.tableWidget.item(i,7).text()
zongfene =float(zongfene) + float(fene)
zongfene = round(zongfene,2)
#获取股票现价
ts.set_token('184fedac133c785faa3197981f94ef25b8b34047471c081bdb6323c8')
pro = ts.pro_api()
df = pro.daily(ts_code=codelist, trade_date=data)
close = df.iloc[0,5]
#现价
ui.tableWidget.setItem(i, 5, QTableWidgetItem(str(close)))
#计算股票总市值（现价）
shizhi = float(close)*float(lastnumber)
zongshizhi =float(zongshizhi) + float(shizhi)
zongshizhi = round(zongshizhi,2)
#盈亏金额
yingkui = (float(close)-float(lastclose))*float(lastnumber)
yingkui =round(yingkui,3)
ui.tableWidget.setItem(i,4,QTableWidgetItem(str(yingkui)))
#盈亏率
yinglv = 100 * yingkui / (float(lastclose) * float(lastnumber))
yinglv = round(yinglv,2)
ui.tableWidget.setItem(i,6,QTableWidgetItem(str(yinglv)))
ui.lineEdit.setText(str(zongfene))
ui.lineEdit.setReadOnly(True)
ui.lineEdit_15.setText(str(zongshizhi))
ui.lineEdit_15.setReadOnly(True)
#总资产=总市值+卖出盈利
yingli = ui.lineEdit_2.text()
zongzichan = float(zongshizhi)+float(yingli)
ui.lineEdit_14.setText(str(zongzichan))
ui.lineEdit_14.setReadOnly(True)
#单位净值
jingzhi = float(zongshizhi)/float(zongfene)
jingzhi = round(jingzhi,2)
ui.lineEdit_13.setText(str(jingzhi))
ui.lineEdit_13.setReadOnly(True)
ui.pushButton_5.clicked.connect(gengxin)
def jisuan():
# 分红送股
n = ui.lineEdit_16.text()
shuliang = ui.lineEdit_5.text()
jiage = ui.lineEdit_4.text()
newshuliang = float(shuliang)*(float(10)+float(n))/float(10)
newjiage = float(jiage)*float(10)/(float(10)+float(n))
ui.lineEdit_18.setText(str(newshuliang))
ui.lineEdit_17.setText(str(newjiage))
ui.pushButton_8.clicked.connect(jisuan)
#buy模块
def buy():
codelist = ui.lineEdit_3.text()
number = ui.lineEdit_5.text()
buyprice = ui.lineEdit_4.text()
shouxufei = ui.lineEdit_9.text()
endprice =float(buyprice)*float(number)
k = pro.stock_basic(ts_code=codelist, fields='ts_code,name,area,industry,list_date')
name = k.iloc[0,1]
for i in range(0,30):
if ui.tableWidget.item(i,1) == None:
#股票名称
ui.tableWidget.setItem(i,0,QTableWidgetItem(str(name)))
#股票代码
ui.tableWidget.setItem(i,1, QTableWidgetItem(codelist))
#持仓数量
ui.tableWidget.setItem(i,2,QTableWidgetItem(number))
#交易价格
ui.tableWidget.setItem(i,3, QTableWidgetItem(buyprice))
#总份额
ui.tableWidget.setItem(i,7, QTableWidgetItem(str(endprice)))
#手续费
ui.tableWidget.setItem(i,8,QTableWidgetItem(str(float(buyprice)*float(number)*float(shouxufei)*0.01)))
break
else:
if codelist == ui.tableWidget.item(i,1).text():
# 持仓数量
backnumber = ui.tableWidget.item(i,2).text()
allnumber = float(number)+float(backnumber)
ui.tableWidget.setItem(i,2,QTableWidgetItem(str(allnumber)))
# 交易价格
backprice = ui.tableWidget.item(i,3).text()
allprice = (float(backprice)*float(backnumber)+float(number)*float(buyprice))/(float(backnumber)+float(number))
ui.tableWidget.setItem(i,3,QTableWidgetItem(str(allprice)))
# 总份额
backendprice = ui.tableWidget.item(i,7).text()
allendprice = float(backendprice)+float(number)*float(buyprice)
ui.tableWidget.setItem(i,7,QTableWidgetItem(str(allendprice)))
# 手续费
backshouxufei = ui.tableWidget.item(i,8).text()
allshouxufei = float(backshouxufei)+float(buyprice) * float(number) * float(shouxufei) * 0.01
ui.tableWidget.setItem(i, 8, QTableWidgetItem(str(allshouxufei)))
break
ui.pushButton_3.clicked.connect(buy)
def sell():
shouxufei = ui.lineEdit_9.text()
codelist1 = ui.lineEdit_6.text()
number1 = ui.lineEdit_8.text()
sellprice = ui.lineEdit_7.text()
for i in range(1,30):
if codelist1 == ui.tableWidget.item(i,1).text():
number = ui.tableWidget.item(i,2).text()
if float(number1) > float(number):
child.show()
else:
# 持仓数量
ui.tableWidget.setItem(i, 2, QTableWidgetItem(str(float(number)-float(number1))))
# 总份额
backendprice = ui.tableWidget.item(i,7).text()
price = ui.tableWidget.item(i, 3).text()
allendprice = float(backendprice) - float(number1) * float(price)
ui.tableWidget.setItem(i,7, QTableWidgetItem(str(allendprice)))
# 手续费
backshouxufei = ui.tableWidget.item(i,8).text()
allshouxufei = float(backshouxufei) + float(sellprice) * float(number1) * float(shouxufei) * 0.01
ui.tableWidget.setItem(i,8,QTableWidgetItem(str(allshouxufei)))
#总盈利
zongyingli = float(number1)*(float(sellprice)-float(price))*float(ui.tableWidget.item(i,3).text())-float(allshouxufei)
if len(ui.lineEdit_2.text()) == 0:
ui.lineEdit_2.setText(str(zongyingli))
else:
backzongyingli = ui.lineEdit_2.text()
ui.lineEdit_2.setText(str(float(zongyingli)+float(backzongyingli)))
ui.lineEdit_2.setReadOnly(True)
break
ui.pushButton_4.clicked.connect(sell)
def baocun():
qian = ui.lineEdit_2.text()
workbook = xlwt.Workbook(encoding='utf-8')
sheet = workbook.add_sheet('数据')
sheet.write(0,0,qian)
for i in range(1,30):
for j in range(0,9):
if ui.tableWidget.item(i,j) == None:
continue
else:
shuju = ui.tableWidget.item(i,j).text()
sheet.write(i,j,shuju)
workbook.save('shuju.xls')
ui.pushButton_2.clicked.connect(baocun)
def duqu():
excel_path = "shuju.xls"
excel = xlrd.open_workbook(excel_path, encoding_override="utf-8")
all_sheet = excel.sheets()
for sheet in all_sheet:
qian = sheet.cell_value(0,0)
sheet_row_mount = sheet.nrows#行
sheet_col_mount = sheet.ncols#列
ui.lineEdit_2.setText(str(qian))
for i in range(1,sheet_row_mount):
for j in range(0,sheet_col_mount):
if sheet.cell_value(i,j) == None:
continue
else:
shuju = sheet.cell_value(i,j)
ui.tableWidget.setItem(i,j, QTableWidgetItem(str(shuju)))
ui.pushButton_9.clicked.connect(duqu)
MainWindow.show()
sys.exit(app.exec_())以上是关于借助Tushare接口调用的股票数据制作简单的股票交易小程序的主要内容，如果未能解决你的问题，请参考以下文章