python--杂识--18.1--pandas数据插入sqlite并进行查询

Chasing__Dreams2025-07-09 13:18
python 复制代码
import pandas as pd
import sqlite3

connection = sqlite3.connect(":memory:")
cursor = connection.cursor()
connection.execute("PRAGMA case_sensitive_like = true;")


file_path = "test.csv"
table_name = "student"
filed_name= "name"


header = pd.read_csv(file_path, nrows=1).to_dict(orient="records")[0]

create_table_sql = f"CREATE TABLE {table_name} (" + \
                   ",".join(["{0} {1}".format(key, value) for key, value in header.items()]) + ")"
connection.execute(create_table_sql)

# 查看表结构
cursor.execute("PRAGMA table_info(student)")
print("字段信息：")
for field in cursor.fetchall():
    print(f"名称：{field[1]:<6} | 类型：{field[2]:<8} | 主键：{field[5]}")

# pd读csv内容
df = pd.read_csv(file_path, encoding="utf-8", skiprows=[1], na_values=["null"], keep_default_na=False, dtype=object)
print(df)
# pd将数据插入sqlite
df.to_sql(table_name, connection, if_exists="append", index=False)

# 查表数据
query = f"select name, age, sex, class from {table_name}"
result = pd.read_sql_query(query, connection)
print(result)
connection.close()