5.25

代码行数：200H

学习内容：完成提交大作业。

学习计划：期末完成及格

from tkinter import Tk, Label, Button, Entry
import xlwt as xlwt
from lxml import etree
import requests
import pandas as pd
import matplotlib.pyplot as plt

def two_color_ball():
    url = entry.get()
    xls = xlwt.Workbook()   # 创建一个新的 Excel 文件
    sheet = xls.add_sheet("two_color_ball", cell_overwrite_ok=True)   # 创建一个名为 "two_color_ball" 的工作表，并允许单元格覆盖写入
    row = 1   # 初始化行数
    # url = "http://m.55123.cn/kjh/ssq-history-120.htm"   定义要爬取的网站 URL
    req = requests.get(url)   # 发送 GET 请求并获取响应数据
    resp = req.content.decode("utf-8")   # 将响应数据转换为 UTF-8 编码格式
    html = etree.HTML(resp)   # 使用 etree 解析 HTML 响应数据
    uls = html.xpath("/html/body/div[@class='wrap']/div[@class='lottery']/ul")   # 查找 HTML 文档中的所有 <ul> 元素，并返回包含这些元素的列表
    dates = []
    nums = []
    for ul in uls:   # 遍历每个 <ul> 元素
        date = ul.xpath("li/span[@class='dates']")   # 获取当前 <ul> 元素下的 <li> 子元素的 <span> 子元素（class="dates"）的文本内容，并返回包含这些元素的列表
        number = ul.xpath("li/a/span[@class='ball lred']/text()")   # 获取当前 <ul> 元素下的 <li> 子元素的 <a> 子元素的 <span> 子元素（class="ball lred"）的文本内容，并返回包含这些元素的列表
        tail_number = ul.xpath("li/a/span[@class='ball lblue']/text()")   # 获取当前 <ul> 元素下的 <li> 子元素的 <a> 子元素的 <span> 子元素（class="ball lblue"）的文本内容，并返回包含这些元素的列表
        dates.append(date[0].text)
        nums.append([number[i] for i in range(len(number))] + [tail_number[0]])
    sheet.write(0, 0, "期数")
    sheet.write(0, 1, "红球1")
    sheet.write(0, 2, "红球2")
    sheet.write(0, 3, "红球3")
    sheet.write(0, 4, "红球4")
    sheet.write(0, 5, "红球5")
    sheet.write(0, 6, "红球6")
    sheet.write(0, 7, "蓝球1")
    for i in range(len(dates)):
        sheet.write(i+1, 0, dates[i])
        sheet.write(i+1, 1, nums[i][0])
        sheet.write(i+1, 2, nums[i][1])
        sheet.write(i+1, 3, nums[i][2])
        sheet.write(i+1, 4, nums[i][3])
        sheet.write(i+1, 5, nums[i][4])
        sheet.write(i+1, 6, nums[i][5])
        sheet.write(i+1, 7, nums[i][6])
    xls.save("two_color_ball.xls")
    data = pd.read_excel("two_color_ball.xls")
    plt.plot(data["期数"], data["红球1"], label="红球1", marker="o")
    plt.plot(data["期数"], data["红球2"], label="红球2", marker="o")
    plt.plot(data["期数"], data["红球3"], label="红球3", marker="o")
    plt.plot(data["期数"], data["红球4"], label="红球4", marker="o")
    plt.plot(data["期数"], data["红球5"], label="红球5", marker="o")
    plt.plot(data["期数"], data["红球6"], label="红球6", marker="o")
    plt.plot(data["期数"], data["蓝球1"], label="蓝球1", marker="o")
    plt.legend()
    plt.show()

root = Tk()
root.title("双色球历史数据分析")
root.geometry("400x200")

label = Label(root, text="请输入要爬取的双色球历史数据 URL：")
label.pack()

entry = Entry(root)
entry.pack()

button = Button(root, text="爬取并分析", command=two_color_ball)
button.pack()

root.mainloop()