Python连接数据库代码
conn = pymysql.connect(
host='localhost',
user='root',
password='password',
database='test_db',
charset='utf8'
)
# 写入数据
try:
with conn.cursor() as cursor:
sql = "INSERT INTO users (user_id, age, gender, income, preference, label) VALUES (%s, %s, %s, %s, %s, %s)"
cursor.executemany(sql, data_with_label)
conn.commit()
print('数据写入成功!')
except Exception as e:
conn.rollback()
print(f'数据写入失败:{e}')
finally:
conn.close()