2024.3.14（周四）进度

复习计算机网络的ppt并完成python的课后作业。

编程时间：一个小时

代码量：75行

import requests
from bs4 import BeautifulSoup as bs
import pandas as pd
from matplotlib import pyplot as plt


def get_rank(url):
    rank = []
    headers = {
        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.54 Safari/537.36 Edg/101.0.1210.3"
    }
    resp = requests.get(url, headers=headers)
    if resp.status_code != 200:
        print(f"Failed to retrieve data from {url}")
        return rank

    soup = bs(resp.content, "lxml")
    rows = soup.select("#content-box > div.rk-table-box > table > tbody > tr")
    for row in rows[:10]:  # 获取前10个排名
        university = row.select_one(".name-cn").text.strip()
        score = row.select("td:nth-child(5)")[0].text.strip()
        rank.append([university, score])
    return rank


total = []
years = range(2020, 2025)
for year in years:
    url = f"https://www.shanghairanking.cn/rankings/bcur/{year}10"
    print(f"Fetching data from: {url}")
    title = ['学校名称', '总分']
    df = pd.DataFrame(get_rank(url), columns=title)
    total.append(df)

    # 画图
    plt.rcParams['font.sans-serif'] = ['SimHei']  # 用来正常显示中文标签
    x = list(df["学校名称"])[::-1]
    y = list(df["总分"])[::-1]
    plt.figure(figsize=(20, 8), dpi=100)
    plt.plot(x, y, label="大学排名", marker='o')
    plt.grid(True, linestyle="--", alpha=0.5)
    plt.xlabel("大学名称")
    plt.ylabel("总分")
    plt.title(f"{year}年软科中国最好大学排名Top10", fontsize=20)
    plt.legend(loc="best")
    plt.savefig(f"{year}.png")
    plt.show()


# 查询功能
def query_ranking():
    while True:
        info = input("请输入要查询的大学名称和年份（格式：大学名称 年份）：")
        try:
            university, year = info.split()
            year = int(year)
            if year in years:
                index = year - 2020
                df = total[index]
                if university in df["学校名称"].values:
                    rank = df[df["学校名称"] == university].index[0] + 1
                    print(f"{university}在{year}年排名第{rank}")
                else:
                    print("很抱歉，没有该学校的排名记录！！！")
            else:
                print("很抱歉，没有该年份的排名记录！！！")
        except ValueError:
            print("输入格式错误，请输入大学名称和年份（格式：大学名称 年份）")

        select = input("请选择以下选项:\n   1.继续查询\n   2.结束查询\n")
        if select == '2':
            break


if __name__ == "__main__":
    query_ranking()