股票 出仓点 及 补仓点预测 python代码

使用python的tkinter自制gui界面

 

代码如下：

import matplotlib.pyplot as plt
import tkinter as tk
from tkinter import *
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2Tk
from matplotlib.figure import Figure
import matplotlib.pyplot as plt

# 用于plt显示正常的中文
plt.rcParams['font.sans-serif'] = ['SimHei']
# 用来正常显示负号
plt.rcParams['axes.unicode_minus'] = False


def stock_predict_increase(buy_price, buy_num, new_price, new_num):
    """
    补仓点
    :param buy_price: 买入价格
    :param buy_num: 买入股数
    :param new_price: 当前价格
    :param new_num: 当前买入股数
    :return:
    now_price  总股本买入价格
    now_num  总股本数
    """
    now_num = buy_num + new_num
    old_cost = buy_price * buy_num
    new_cost = new_price * new_num
    cost = old_cost + new_cost

    now_price = cost / now_num
    return now_price, now_num


def stock_predict_decrease(buy_price, buy_num, new_price, new_num):
    """
    出仓点
    :param buy_price: 买入价格
    :param buy_num: 买入股数
    :param new_price: 当前价格
    :param new_num: 当前买出股数
    :return:
    now_price  总股本买入价格
    now_num  总股本数
    """
    now_num = buy_num - new_num
    old_cost = buy_price * buy_num
    new_gain = new_price * new_num
    cost = old_cost - new_gain

    if now_num <= 0:
        now_price = 0
    else:
        now_price = cost / now_num
    return now_price, now_num


def stock_trend_point(buy_price, buy_num, new_price, flag):
    now_price_list = []
    new_num_list = list(range(0, 1000, 100))
    for new_num in range(0, 1000, 100):
        if flag:
            now_price, now_num = stock_predict_increase(buy_price, buy_num, new_price, new_num)
        else:
            now_price, now_num = stock_predict_decrease(buy_price, buy_num, new_price, new_num)
        now_price_list.append(now_price)
    return new_num_list, now_price_list


def _quit():
    """点击退出按钮时调用这个函数"""
    global window
    window.quit()  # 结束主循环
    window.destroy()  # 销毁窗口


def stock_gui(buy_price, buy_num, new_price, flag):
    new_num_list, now_price_list = stock_trend_point(buy_price, buy_num, new_price, flag)

    figure.clf()
    a = figure.add_subplot(111)  # 添加子图:1行1列第1个
    a.plot(new_num_list, now_price_list)
    if flag:
        a.set_xlabel("买入数量变化")
    else:
        a.set_xlabel("卖出数量变化")
    a.set_ylabel("本仓总体价格变化")
    a.set_title('股票价格随买入数量的变化关系')

    list_1 = Listbox(window)  # 创建两个列表组件
    list_2 = Listbox(window)
    for item in new_num_list:  # 第一个小部件插入数据
        list_1.insert(0, item)

    for item in now_price_list:  # 第二个小部件插入数据
        list_2.insert(0, item)

    # list_1.pack()  # 将小部件放置到主窗口中
    # list_2.pack()

    # 将绘制的图形显示到tkinter,并创建属于window的canvas画布,将图f置于画布上
    canvas = FigureCanvasTkAgg(figure, master=window)
    canvas.draw()
    canvas.get_tk_widget().place(x=100, y=100, anchor='nw')


window = tk.Tk()
window.title('stock predict')
window.geometry('900x600')

figure = plt.figure(figsize=(5, 4), dpi=100)

stock_num = 1000
stock_price = 30
stock_new_price = 30

var_string_1 = tk.StringVar()
var_string_1.set('当前股票价格：' + str(stock_price))
var_string_2 = tk.StringVar()
var_string_2.set('当前股票股数：' + str(stock_num))
var_string_3 = tk.StringVar()
var_string_3.set('新股票价格：' + str(stock_new_price))
var_string_4 = tk.StringVar()


def price_control(flag):
    global stock_price
    if flag:
        stock_price = stock_price + 1
        var_string_1.set('当前股票价格：' + str(stock_price))
    else:
        stock_price = stock_price - 1
        var_string_1.set('当前股票价格：' + str(stock_price))


def number_control(flag):
    global stock_num
    if flag:
        stock_num = stock_num + 100
        var_string_2.set('当前股票股数：' + str(stock_num))
    else:
        stock_num = stock_num - 100
        var_string_2.set('当前股票股数：' + str(stock_num))


def new_price_control(flag):
    global stock_new_price
    if flag:
        stock_new_price = stock_new_price + 1
        var_string_3.set('新股票价格：' + str(stock_new_price))
    else:
        stock_new_price = stock_new_price - 1
        var_string_3.set('新股票价格：' + str(stock_new_price))


label1 = tk.Label(window, text='股票预测信息', anchor='nw')
label1.place(x=370, y=0, anchor='nw')

button1 = tk.Button(window, text='买入预测 ', command=lambda: stock_gui(stock_price, stock_num, stock_new_price, True))
button1.place(x=100, y=50, anchor='nw')

button2 = tk.Button(window, text='卖出预测 ', command=lambda: stock_gui(stock_price, stock_num, stock_new_price, False))
button2.place(x=200, y=50, anchor='nw')

button3 = tk.Button(window, text='当前价格增', command=lambda: price_control(True))
button3.place(x=300, y=50, anchor='nw')

button4 = tk.Button(window, text='当前价格减', command=lambda: price_control(False))
button4.place(x=400, y=50, anchor='nw')

button5 = tk.Button(window, text='当前股数增', command=lambda: number_control(True))
button5.place(x=500, y=50, anchor='nw')

button6 = tk.Button(window, text='当前股数减', command=lambda: number_control(False))
button6.place(x=600, y=50, anchor='nw')

button7 = tk.Button(window, text='新股价格增', command=lambda: new_price_control(True))
button7.place(x=700, y=50, anchor='nw')

button8 = tk.Button(window, text='新股价格减', command=lambda: new_price_control(False))
button8.place(x=800, y=50, anchor='nw')

# 创建一个按钮,并把上面那个函数绑定过来
button10 = tk.Button(master=window, text="退出", command=_quit)
# 按钮放在下边
button10.place(x=460, y=550, anchor='nw')

# text1 = tk.Text(window)
# text1.insert()

label2 = tk.Label(window, textvariable=var_string_1, anchor='nw')
label2.place(x=700, y=250, anchor='nw')

label3 = tk.Label(window, textvariable=var_string_2, anchor='nw')
label3.place(x=700, y=300, anchor='nw')

label4 = tk.Label(window, textvariable=var_string_3, anchor='nw')
label4.place(x=700, y=350, anchor='nw')

window.mainloop()  # 进入消息循环

# if __name__ == "__main__":