import pandas as pd
import sqlite3

# ===================== 1. 连接数据库 =====================
# 直接连接你之前生成的 college_info.db 数据库文件
conn = sqlite3.connect("college_info.db")

# ===================== 2. 查看数据库里有哪些表（必看！） =====================
# 运行这行，就能看到你存的所有表名（college2024、college2023...）
cursor = conn.cursor()
cursor.execute("SELECT name FROM sqlite_master WHERE type='table';")
table_list = cursor.fetchall()
print("📊 数据库里的所有表：", table_list)

# ===================== 3. 读取单张表（比如2024年的数据） =====================
df_2024 = pd.read_sql("SELECT * FROM college2024", conn)
print("\n📄 2024年高考分数线数据：")
print(df_2024)

# ===================== 4. 批量读取所有年份表（2020-2024） =====================
print("\n📈 批量读取所有年份数据：")
for year in [2020,2021,2022,2023,2024]:
    df = pd.read_sql(f"SELECT * FROM college{year}", conn)
    print(f"\n===== {year}年数据 =====")
    print(df)

# ===================== 关闭连接 =====================
conn.close()