PostgreSQLのデータをpandasに読み込んでみる

ヒノマルク

今日はpostgreSQLのデータをpandasのデータフレームに読み込めるようにします。
データベースからCSVに吐き出したものをpandasに読み込んでいる方はぜひ直接DBから読み込んでみてください。

psycopg2をインポート

import psycopg2

コネクションを作成

conn = psycopg2.connect("dbname=postgres user=postgres password=secret host=localhost port=5432")

pandasのread_sqlメソッドを使ってデータをpostgreSQLからdataframeに読み込む

import pandas as pd
# テーブル名一覧を取得するSQL
query = """SELECT tablename FROM pg_catalog.pg_tables"""
df = pd.read_sql(query, con=conn)

dataframeに読み込んだデータを確認

# head()メソッドで上から5件を表示する
df.head()

Out[0]




  
    
      
      tablename
    
  
  
    
      0
      test
    
    
      1
      pg_statistic
    
    
      2
      pg_type
    
    
      3
      pg_foreign_table
    
    
      4
      pg_authid

	tablename
0	test
1	pg_statistic
2	pg_type
3	pg_foreign_table
4	pg_authid

cx_Oracleとほとんど同じでした。
コネクションさえ作成できればpandasのread_sql()メソッドでデータを読み込めてしまうので楽ですね。