Запросы к реляционным базам данных в Python

Введение в импорт данных в Python

Hugo Bowne-Anderson

Data Scientist at DataCamp

Базовый SQL-запрос

SELECT * FROM Table_Name

Возвращает все столбцы и строки таблицы
Пример:

SELECT * FROM Orders

Будем использовать SQLAlchemy и pandas

Процесс выполнения SQL-запроса

Импорт пакетов и функций
Создание движка базы данных
Подключение к движку
Запрос к базе данных
Сохранение результатов в DataFrame
Закрытие соединения

Ваш первый SQL-запрос

from sqlalchemy import create_engine
import pandas as pd

engine = create_engine('sqlite:///Northwind.sqlite')

con = engine.connect()

rs = con.execute("SELECT * FROM Orders")

df = pd.DataFrame(rs.fetchall())

con.close()

Вывод результатов запроса

print(df.head())

       0      1   2                      3                      4 
0  10248  VINET   5   7/4/1996 12:00:00 AM   8/1/1996 12:00:00 AM   
1  10251  VICTE   3   7/8/1996 12:00:00 AM   8/5/1996 12:00:00 AM   
2  10254  CHOPS   5  7/11/1996 12:00:00 AM   8/8/1996 12:00:00 AM   
3  10256  WELLI   3  7/15/1996 12:00:00 AM  8/12/1996 12:00:00 AM   
4  10258  ERNSH   1  7/17/1996 12:00:00 AM  8/14/1996 12:00:00 AM

Задание имён столбцов DataFrame

from sqlalchemy import create_engine
import pandas as pd
engine = create_engine('sqlite:///Northwind.sqlite')
con = engine.connect()
rs = con.execute("SELECT * FROM Orders")
df = pd.DataFrame(rs.fetchall())
df.columns = rs.keys()
con.close()

Задание имён столбцов DataFrame

print(df.head())

   OrderID CustomerID  EmployeeID              OrderDate  
0    10248      VINET           5   7/4/1996 12:00:00 AM   
1    10251      VICTE           3   7/8/1996 12:00:00 AM   
2    10254      CHOPS           5  7/11/1996 12:00:00 AM   
3    10256      WELLI           3  7/15/1996 12:00:00 AM   
4    10258      ERNSH           1  7/17/1996 12:00:00 AM

Использование контекстного менеджера

from sqlalchemy import create_engine
import pandas as pd
engine = create_engine('sqlite:///Northwind.sqlite')

with engine.connect() as con:
    rs = con.execute("SELECT OrderID, OrderDate, ShipName FROM Orders")
    df = pd.DataFrame(rs.fetchmany(size=5))
    df.columns = rs.keys()

Давайте потренируемся!

Введение в импорт данных в Python