virtualenv -p python3 venvConnect to Postgresql with Python 3.x and get Pandas Dataframe
Upasana | February 13, 2020 | 2 min read | 656 views
We will be learning
- 
How to connect to PostgreSQL in Python 
- 
How to get output of query as Pandas Dataframe 
Connect to PostgreSQL in Python
We will be using PyGreSQL Library to make connection.
Install PyGreSQL
Create virtual environment in project directory
Activate virtual environment
source venv/bin/activateTo know more about virtual environment, click here
Install package
pip install PyGreSQLDefine a class for connecting to PostgreSQL
from pgdb import connect
class PostGresql(object):
    def __init__(self):
        self.connection_pool = None
        self.database = "database" //(1)
        self.host = "host" //(2)
        self.user = "user" //(3)
        self.password = "password" //(4)
    def get_connection(self):
        connection = connect(database=self.database,host=self.host,user=self.user,password=self.password)
        return connection- 
Database Name 
- 
Host of database 
- 
Username 
- 
Password of DB 
Now, we will be making another class to initiate the connection
import pandas as pd
class PostgreDao(object):
    def __init__(self, postSqlDb):
        self.postSqlDb = postSqlDb
        print("Acquiring connection")
        self.connection = postSqlDb.get_connection()
    def __del__(self):
        print("Closing db connection")
        self.connection.close()
    def read_table(self, query):
        """Read SQL Table
            Args:
              query: text, query to read table or multiple tables
            Returns:
              pandas dataframe, if query is valid
            """
        df = pd.read_sql_query(query, self.connection)
        return dfmain.py
dbcon = PostGresql()
dbclass = PostgreDao(dbcon)
query = """
            select * from table
            """
data = dbclass.read_table(query)Output of data will be a pandas dataframe
In case, you are looking to connect to MySQL Database then follow this article
Top articles in this category:
- Connect to MySQL with Python 3.x and get Pandas Dataframe
- Connect to Cassandra with Python 3.x and get Pandas Dataframe
- Google Colab: import data from google drive as pandas dataframe
- Top 100 interview questions on Data Science & Machine Learning
- Google Data Scientist interview questions with answers
- Python - Get Google Analytics Data
- Python send GMAIL with attachment
Recommended books for interview preparation:
                    Book you may be interested in..
                
                
            
                    Book you may be interested in..
                
                
             
                         
                        