读取csv文件中的固定列写入到数据库

import csv
import mysql.connector

# 打开 CSV 文件并读取数据
with open('data.csv', 'r') as file:
    csv_data = list(csv.reader(file))

# 打印 CSV 文件内容和提取需要的列的值
headers = csv_data[0]
for row in csv_data[1:]:
    step_id = row[0]
    part_id = row[1]
    for i in range(2, 7):  # 提取需要的列的值
        part_key = headers[i]
        if part_key.startswith("step_"):
            part_value = row[i]
            print(f"step_id: {step_id}, part_id: {part_id}")
            print(f"part_key: {part_key}, part_value: {part_value}")

            # 将数据插入到 MySQL 表中
            # 创建 MySQL 连接
            cnx = mysql.connector.connect(
                host='localhost',
                user='your_username',
                password='your_password',
                database='your_database'
            )
            cursor = cnx.cursor()

            # 执行插入数据的 SQL 语句
            sql = "INSERT INTO your_table (step_id, part_id, part_key, part_value) VALUES (%s, %s, %s, %s)"
            values = (step_id, part_id, part_key, part_value)
            cursor.execute(sql, values)

            # 提交并关闭数据库连接
            cnx.commit()
            cursor.close()
            cnx.close()