Consultando bancos relacionais em Python

Introdução à importação de dados em Python

Hugo Bowne-Anderson

Data Scientist at DataCamp

Consulta SQL básica

SELECT * FROM Table_Name

Retorna todas as colunas de todas as linhas da tabela
Exemplo:

SELECT * FROM Orders

Vamos usar SQLAlchemy e pandas

Fluxo de uma consulta SQL

Importar pacotes e funções
Criar o engine do banco
Conectar ao engine
Consultar o banco
Salvar resultados em um DataFrame
Fechar a conexão

Sua primeira consulta 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()

Imprimir os resultados da consulta

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

Definir nomes das colunas do 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()

Definir nomes das colunas do data frame

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

Usando o gerenciador de contexto

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()

Vamos praticar!

Introdução à importação de dados em Python