python-panda e database come mysql


97

La documentazione di Pandas contiene numerosi esempi di best practice per lavorare con i dati archiviati in vari formati.

Tuttavia, non sono in grado di trovare buoni esempi per lavorare con database come MySQL, ad esempio.

Qualcuno può indicarmi dei collegamenti o fornire alcuni frammenti di codice su come convertire i risultati delle query utilizzando mysql-python in frame di dati in Panda in modo efficiente?




Dai anche un'occhiata a Blaze .
osa

Se sei disposto a spendere soldi, credo che il libro di Wes McKinney ("Python for Data Analysis") contenga alcuni esempi utili.
MTrenfield

Risposte:


102

Come dice Wes, read_sql di io / sql lo farà, una volta ottenuta una connessione al database utilizzando una libreria compatibile DBI. Possiamo guardare due brevi esempi usando le librerie MySQLdbe cx_Oracleper connetterci a Oracle e MySQL e interrogare i loro dizionari di dati. Ecco l'esempio per cx_Oracle:

import pandas as pd
import cx_Oracle

ora_conn = cx_Oracle.connect('your_connection_string')
df_ora = pd.read_sql('select * from user_objects', con=ora_conn)    
print 'loaded dataframe from Oracle. # Records: ', len(df_ora)
ora_conn.close()

Ed ecco l'esempio equivalente per MySQLdb:

import MySQLdb
mysql_cn= MySQLdb.connect(host='myhost', 
                port=3306,user='myusername', passwd='mypassword', 
                db='information_schema')
df_mysql = pd.read_sql('select * from VIEWS;', con=mysql_cn)    
print 'loaded dataframe from MySQL. records:', len(df_mysql)
mysql_cn.close()

57

Per i lettori recenti di questa domanda: i panda hanno il seguente avviso nei loro documenti per la versione 14.0 :

Avviso: alcune delle funzioni o degli alias di funzioni esistenti sono stati deprecati e verranno rimossi nelle versioni future. Ciò include: tquery, uquery, read_frame, frame_query, write_frame.

E:

Avviso: il supporto per il sapore "mysql" quando si utilizzano oggetti di connessione DBAPI è stato deprecato. MySQL sarà ulteriormente supportato con i motori SQLAlchemy (GH6900).

Questo rende molte delle risposte qui obsolete. Dovresti usare sqlalchemy:

from sqlalchemy import create_engine
import pandas as pd
engine = create_engine('dialect://user:pass@host:port/schema', echo=False)
f = pd.read_sql_query('SELECT * FROM mytable', engine, index_col = 'ID')

il caricamento di una tabella con 133 righe e 7 colonne richiede circa 30 secondi .. puoi fornire alcuni spunti sul perché?
idoda

@idoda [in generale questo non è l'argomento della domanda ed è meglio fare una nuova domanda in modo da ottenere più opinioni]. Sei sicuro che non sia una questione di ritardo nella richiesta? L'invio della query e il recupero dei risultati sono semplicemente molto più rapidi?
Korem

@Korem Ho pensato di aprirne uno nuovo, ma prima volevo assicurarmi che non fosse banale. Quando utilizzo un client mySql (Sequel pro) e interrogo il database, i risultati vengono visualizzati molto più velocemente. Quando dici "semplicemente inviare e poi recuperare", è questo che intendi? (utilizzando un client)
idoda

@idoda intendo confrontare il tempo necessario per l'esecuzione engine.execute("select * FROM mytable")con il tempo necessario per l'esecuzionepd.read_sql_query('SELECT * FROM mytable', engine)
Korem

Si può passare una query sqlalchemy (session.query come nella mia risposta di seguito) direttamente a un metodo pandas? Sarebbe uno squartatore!
dmvianna

23

Per la cronaca, ecco un esempio che utilizza un database sqlite:

import pandas as pd
import sqlite3

with sqlite3.connect("whatever.sqlite") as con:
    sql = "SELECT * FROM table_name"
    df = pd.read_sql_query(sql, con)
    print df.shape

1
È possibile specificare la colonna da utilizzare come indice specificando index_col='timestamp'in frame_query.
Lumaca meccanica

19

Preferisco creare query con SQLAlchemy e quindi creare un DataFrame da esso. SQLAlchemy rende più facile combinare le condizioni SQL in modo Python se si intende mescolare e abbinare le cose più e più volte.

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Table
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from pandas import DataFrame
import datetime

# We are connecting to an existing service
engine = create_engine('dialect://user:pwd@host:port/db', echo=False)
Session = sessionmaker(bind=engine)
session = Session()
Base = declarative_base()

# And we want to query an existing table
tablename = Table('tablename', 
    Base.metadata, 
    autoload=True, 
    autoload_with=engine, 
    schema='ownername')

# These are the "Where" parameters, but I could as easily 
# create joins and limit results
us = tablename.c.country_code.in_(['US','MX'])
dc = tablename.c.locn_name.like('%DC%')
dt = tablename.c.arr_date >= datetime.date.today() # Give me convenience or...

q = session.query(tablename).\
            filter(us & dc & dt) # That's where the magic happens!!!

def querydb(query):
    """
    Function to execute query and return DataFrame.
    """
    df = DataFrame(query.all());
    df.columns = [x['name'] for x in query.column_descriptions]
    return df

querydb(q)

Inoltre è necessario specificare il driver se non è uguale a quello predefinito di SQLAlchemy :dialect+driver://user:pwd@host:port/db
Nuno André

11

Esempio MySQL:

import MySQLdb as db
from pandas import DataFrame
from pandas.io.sql import frame_query

database = db.connect('localhost','username','password','database')
data     = frame_query("SELECT * FROM data", database)

7
frame_queryè ora deprecato. Ora usa pd.read_sql(query, db)invece.
Robert Smith

8

La stessa sintassi funziona anche per Ms SQL Server utilizzando podbc.

import pyodbc
import pandas.io.sql as psql

cnxn = pyodbc.connect('DRIVER={SQL Server};SERVER=servername;DATABASE=mydb;UID=username;PWD=password') 
cursor = cnxn.cursor()
sql = ("""select * from mytable""")

df = psql.frame_query(sql, cnxn)
cnxn.close()

5

Ed è così che ti connetti a PostgreSQL usando il driver psycopg2 (installa con "apt-get install python-psycopg2" se sei su un sistema operativo derivato Debian Linux).

import pandas.io.sql as psql
import psycopg2

conn = psycopg2.connect("dbname='datawarehouse' user='user1' host='localhost' password='uberdba'")

q = """select month_idx, sum(payment) from bi_some_table"""

df3 = psql.frame_query(q, conn)


4

pandas.io.sql.frame_queryè deprecato. Usa pandas.read_sqlinvece.


1

importa il modulo

import pandas as pd
import oursql

Collegare

conn=oursql.connect(host="localhost",user="me",passwd="mypassword",db="classicmodels")
sql="Select customerName, city,country from customers order by customerName,country,city"
df_mysql = pd.read_sql(sql,conn)
print df_mysql

Funziona perfettamente e utilizza pandas.io.sql frame_works (con l'avviso di deprecazione). Il database utilizzato è il database di esempio dal tutorial mysql.


0

Questo dovrebbe funzionare bene.

import MySQLdb as mdb
import pandas as pd
con = mdb.connect(‘127.0.0.1’, root’, password’, database_name’);
with con:
 cur = con.cursor()
 cur.execute(“select random_number_one, random_number_two, random_number_three from randomness.a_random_table”)
 rows = cur.fetchall()
 df = pd.DataFrame( [[ij for ij in i] for i in rows] )
 df.rename(columns={0: Random Number One’, 1: Random Number Two’, 2: Random Number Three’}, inplace=True);
 print(df.head(20))

0

Questo mi ha aiutato a connettermi ad AWS MYSQL (RDS) dalla funzione lambda basata su python 3.x e caricare in un DataFrame panda

import json
import boto3
import pymysql
import pandas as pd
user = 'username'
password = 'XXXXXXX'
client = boto3.client('rds')
def lambda_handler(event, context):
    conn = pymysql.connect(host='xxx.xxxxus-west-2.rds.amazonaws.com', port=3306, user=user, passwd=password, db='database name', connect_timeout=5)
    df= pd.read_sql('select * from TableName limit 10',con=conn)
    print(df)
    # TODO implement
    #return {
    #    'statusCode': 200,
    #    'df': df
    #}

0

Per gli utenti di Postgres

import psycopg2
import pandas as pd

conn = psycopg2.connect("database='datawarehouse' user='user1' host='localhost' password='uberdba'")

customers = 'select * from customers'

customers_df = pd.read_sql(customers,conn)

customers_df

1
Potresti sottolineare la differenza nella risposta di @Will e perché la tua soluzione dovrebbe essere scelta?
Sebastian
Utilizzando il nostro sito, riconosci di aver letto e compreso le nostre Informativa sui cookie e Informativa sulla privacy.
Licensed under cc by-sa 3.0 with attribution required.