diff --git a/mysqlconnect.py b/mysqlconnect.py new file mode 100644 index 0000000..14dd650 --- /dev/null +++ b/mysqlconnect.py @@ -0,0 +1,38 @@ +import os +import csv +import mysql.connector + +def import_data_from_csv_to_mysql(): + # 设置MySQL连接参数 + mysql_host = "localhost" + mysql_user = "root" + mysql_password = "123456" + mysql_database = "data" + + # 连接MySQL数据库 + conn = mysql.connector.connect(host=mysql_host, user=mysql_user, password=mysql_password, database=mysql_database) + + # 获取MySQL游标 + cursor = conn.cursor() + + # 准备SQL语句 + sql = "INSERT INTO" " congtent (title,comment_num,price, is_self_support,args0,args1,args2,totalCount,picFlagCount,five_fourStarCount,two_threeStarCount,oneStarCount,againCount,comments,datas,users,styles,colors,label)" " VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)" + + # 打开CSV文件并读取数据 + with open('congtent.csv', encoding='utf-8') as csv_file: + csv_reader = csv.reader(csv_file, delimiter=',') + next(csv_reader) + for row in csv_reader: + # 插入数据到MySQL数据库 + cursor.execute(sql, tuple(row)) + + # 提交数据到MySQL数据库 + conn.commit() + + # 关闭连接和游标 + cursor.close() + conn.close() + +if __name__ == '__main__': + import_data_from_csv_to_mysql() + print("CSV数据导入MySQL数据库成功!")