Python 函数：将 txt、csv 和 xlsx 文件导入 MySQL 数据库

本文介绍了三个不同的函数，用于将 txt 文件、csv 文件和 xlsx 文件存储到 MySQL 数据库中。

1. txt_to_mysql 函数用于将 txt 文件存储到 MySQL 数据库中：

import mysql.connector

def txt_to_mysql(filename, db_config):
    conn = mysql.connector.connect(**db_config)
    cursor = conn.cursor()

    with open(filename, 'r') as file:
        for line in file:
            # 处理每一行数据并插入到数据库
            data = line.strip().split('	')
            query = 'INSERT INTO table_name (column1, column2, column3) VALUES (%s, %s, %s)'
            values = (data[0], data[1], data[2])
            cursor.execute(query, values)

    conn.commit()
    cursor.close()
    conn.close()

2. csv_to_mysql 函数用于将 csv 文件存储到 MySQL 数据库中：

import csv
import mysql.connector

def csv_to_mysql(filename, db_config):
    conn = mysql.connector.connect(**db_config)
    cursor = conn.cursor()

    with open(filename, 'r') as file:
        csv_data = csv.reader(file)
        next(csv_data)  # 跳过标题行
        for row in csv_data:
            # 处理每一行数据并插入到数据库
            query = 'INSERT INTO table_name (column1, column2, column3) VALUES (%s, %s, %s)'
            values = (row[0], row[1], row[2])
            cursor.execute(query, values)

    conn.commit()
    cursor.close()
    conn.close()

3. excel_to_mysql 函数用于将 xlsx 文件存储到 MySQL 数据库中（需要使用 openpyxl 库）：

import openpyxl
import mysql.connector

def excel_to_mysql(filename, db_config):
    conn = mysql.connector.connect(**db_config)
    cursor = conn.cursor()

    workbook = openpyxl.load_workbook(filename)
    sheet = workbook.active

    for row in sheet.iter_rows(min_row=2):  # 跳过标题行
        # 处理每一行数据并插入到数据库
        query = 'INSERT INTO table_name (column1, column2, column3) VALUES (%s, %s, %s)'
        values = (row[0].value, row[1].value, row[2].value)
        cursor.execute(query, values)

    conn.commit()
    cursor.close()
    conn.close()

请注意替换代码中的 table_name 为你的数据库表名，并根据需要修改列名和处理每一行数据的逻辑。另外，db_config 是一个包含数据库连接信息的字典，例如：

db_config = {
    'host': 'localhost',
    'database': 'database_name',
    'user': 'username',
    'password': 'password',
    'charset': 'utf8'
}