Created
November 7, 2019 23:18
-
-
Save amitrani6/24e4bafb344d9ed184755d0d5af81e01 to your computer and use it in GitHub Desktop.
A Python script that connects to a PostgreSQL database with an initialization and configuration file to create a Pandas data frame.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# This Python script connects to a PostgreSQL database and utilizes Pandas to obtain data and create a data frame | |
# A initialization and configuration file is used to protect the author's login credentials | |
import psycopg2 | |
import pandas as pd | |
# Import the 'config' funtion from the config.py file | |
from config import config | |
# Establish a connection to the database by creating a cursor object | |
# Obtain the configuration parameters | |
params = config() | |
# Connect to the PostgreSQL database | |
conn = psycopg2.connect(**params) | |
# Create a new cursor | |
cur = conn.cursor() | |
# A function that takes in a PostgreSQL query and outputs a pandas database | |
def create_pandas_table(sql_query, database = conn): | |
table = pd.read_sql_query(sql_query, database) | |
return table | |
# Utilize the create_pandas_table function to create a Pandas data frame | |
# Store the data as a variable | |
vendor_info = create_pandas_table("SELECT vendor_id, vendor_name FROM vendors ORDER BY vendor_name") | |
vendor_info | |
# Close the cursor and connection to so the server can allocate | |
# bandwidth to other requests | |
cur.close() | |
conn.close() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment