PythonでCSVデータをデータベースに登録する方法

Pythonはデータ分析やウェブ開発など、様々な用途で使用されています。その中でも、CSVデータをデータベースに登録するというタスクは、データ分析やデータ管理において非常に重要なスキルとなります。この記事では、Pythonを使用してCSVデータをデータベースに登録する方法を解説します。

CSVデータの読み込み

まずは、PythonでCSVデータを読み込む方法から始めましょう。Pythonの標準ライブラリであるcsvモジュールを使用します。

import csv

with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        print(row)

このコードは、data.csvというCSVファイルを読み込み、その内容を行ごとに表示します。

データベースへの接続

次に、Pythonからデータベースに接続する方法を見ていきましょう。ここでは、MySQLを例に取ります。

import mysql.connector

cnx = mysql.connector.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='your_database'
)

このコードは、指定したホスト、ユーザー名、パスワード、データベースを使用してMySQLデータベースに接続します。

CSVデータのデータベースへの登録

最後に、読み込んだCSVデータをデータベースに登録する方法を見ていきましょう。

import csv
import mysql.connector

cnx = mysql.connector.connect(
    host='localhost',
    user='your_username',
    password='your_password',
    database='your_database'
)
cursor = cnx.cursor()

with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
        cursor.execute(
            "INSERT INTO your_table (column1, column2, ...) VALUES (%s, %s, ...)",
            row
        )

cnx.commit()

このコードは、CSVデータを読み込み、そのデータを指定したテーブルに登録します。登録が完了したら、commitメソッドを呼び出して、変更をデータベースに反映します。

以上が、Pythonを使用してCSVデータをデータベースに登録する基本的な方法です。この方法をマスターすれば、大量のデータを効率的に管理することが可能となります。.