Как импортировать CSV в SQLite с помощью Python? 📊

import sqlite3
import csv

# Устанавливаем соединение с базой данных SQLite
conn = sqlite3.connect('mydatabase.db')
cursor = conn.cursor()

# Создаем таблицу в базе данных
cursor.execute('''CREATE TABLE IF NOT EXISTS mytable (column1, column2, column3)''')

# Открываем CSV файл и читаем его содержимое
with open('myfile.csv', 'r') as file:
    csv_data = csv.reader(file)
    next(csv_data)  # Пропускаем заголовок

    # Импортируем данные в базу данных
    cursor.executemany('INSERT INTO mytable VALUES (?, ?, ?)', csv_data)

# Сохраняем изменения и закрываем соединение с базой данных
conn.commit()
conn.close()
   

Как импортировать CSV в SQLite Python

Импортирование данных из CSV-файлов в базу данных SQLite с использованием языка программирования Python - очень полезная и распространенная задача, которую можно легко выполнить с помощью нескольких шагов. В этой статье мы рассмотрим весь процесс с примерами кода.

1. Установка модуля SQLite3

Первым шагом является установка модуля SQLite3. Убедитесь, что вы уже установили Python на своем компьютере. Модуль SQLite3 входит в стандартную библиотеку Python, поэтому установка не требуется.

2. Создание базы данных SQLite

Прежде чем импортировать CSV-файл, мы должны создать базу данных SQLite. Вот пример кода для создания базы данных:

import sqlite3

# Создание подключения к базе данных
conn = sqlite3.connect('mydatabase.db')

# Создание курсора
cursor = conn.cursor()

# Создание таблицы
cursor.execute('''CREATE TABLE IF NOT EXISTS mytable (
                    column1 TEXT,
                    column2 TEXT,
                    column3 INTEGER
                )''')

# Сохранение изменений
conn.commit()

# Закрытие соединения
conn.close()
    

В этом примере мы создаем базу данных под названием "mydatabase.db" и таблицу под названием "mytable". Таблица имеет три столбца: "column1" с типом TEXT, "column2" с типом TEXT и "column3" с типом INTEGER.

3. Чтение и импортирование CSV-файла

Теперь давайте рассмотрим код, который читает CSV-файл и импортирует его данные в базу данных SQLite:

import csv
import sqlite3

# Открытие CSV-файла
with open('data.csv', 'r') as file:
    # Создание объекта csv.reader
    csv_data = csv.reader(file)
    
    # Подключение к базе данных SQLite
    conn = sqlite3.connect('mydatabase.db')
    cursor = conn.cursor()
    
    # Итерация по строкам CSV-файла
    for row in csv_data:
        # Вставка данных в базу данных
        cursor.execute("INSERT INTO mytable (column1, column2, column3) VALUES (?, ?, ?)", tuple(row))
    
    # Сохранение изменений
    conn.commit()
    
    # Закрытие соединения
    conn.close()
    

В этом коде мы открываем CSV-файл с помощью функции `open()`. Затем мы создаем объект `csv_reader` с помощью модуля `csv`, который позволяет нам читать данные из файла.

После этого мы устанавливаем соединение с базой данных SQLite и создаем курсор для выполнения операций. Затем мы проходим по каждой строке CSV-файла и вставляем данные в таблицу базы данных с помощью метода `execute()`.

4. Проверка импортированных данных

Чтобы убедиться, что данные были успешно импортированы, мы можем выполнить запрос SELECT для извлечения данных из таблицы. Вот пример кода:

import sqlite3

# Подключение к базе данных SQLite
conn = sqlite3.connect('mydatabase.db')
cursor = conn.cursor()

# Выполнение запроса SELECT
cursor.execute("SELECT * FROM mytable")

# Получение всех результатов
rows = cursor.fetchall()

# Вывод результатов
for row in rows:
    print(row)

# Закрытие соединения
conn.close()
    

Этот код выполняет простой запрос SELECT для выбора всех строк из таблицы "mytable". Затем мы проходим по каждой строке и выводим их в консоль.

Заключение

В этой статье мы рассмотрели процесс импортирования CSV-файла в базу данных SQLite с использованием языка программирования Python. Мы создали базу данных, таблицу, прочитали CSV-файл и импортировали его данные в базу данных. Теперь вы знаете, как выполнить эту задачу в своих проектах.