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()