How to import and export data using CSV files in PostgreSQL
Last Updated :
23 Sep, 2021
In this article, we are going to see how to import and export data using CSV file in PostgreSQL, the data in CSV files can be easily imported and exported using PostgreSQL.
To create a CSV file, open any text editor (notepad, vim, atom). Write the column names in the first line. Add row values separated by commas to the corresponding columns in the next lines. Save the file as Demo_data.csv as shown below.
Now create a new table using psycopg2 where we will import the data from the CSV file:
Python3
import psycopg2
# connection establishment
conn = psycopg2.connect(
database="postgres",
user='postgres',
password='password',
host='localhost',
port='5432'
)
conn.autocommit = True
# Creating a cursor object
cursor = conn.cursor()
# query to create a table
sql = ''' CREATE TABLE demo
(
id INT,
name VARCHAR(50),
city VARCHAR(50),
age INT
); '''
# executing above query
cursor.execute(sql)
print("Table has been created successfully!!")
# Closing the connection
conn.close()
Output:
Table has been created successfully!!
Importing data using a CSV file
We use COPY command with FROM keyword to import the contents of the CSV file into a new table.
Syntax: COPY <table name> FROM 'location + file_name' DELIMITER ',' CSV HEADER;
Where:
- <table name> – name of the table where we want to import data.
- ‘location + file_name’ – full path of the CSV file from which we are importing data (make sure you have 'read' access to the file).
- DELIMITER ‘,’ – specifies the delimiter which is comma (,)
- CSV – specifies the format of the file from which we are importing data.
- HEADER – specifies that we have a header row in our .csv file and while importing we should ignore the first row.
Example:
Python3
import psycopg2
# connection establishment
conn = psycopg2.connect(
database="postgres",
user='postgres',
password='password',
host='localhost',
port='5432'
)
conn.autocommit = True
# Creating a cursor object
cursor = conn.cursor()
# query to import data from given csv
sql = '''COPY demo FROM
'C:\\Users\\DELL\\Downloads\\Demo_data.csv'
DELIMITER ',' CSV HEADER'''
# executing above query
cursor.execute(sql)
# Display the table
cursor.execute('SELECT * FROM demo')
print(cursor.fetchall())
# Closing the connection
conn.close()
Output:
[(1, 'Harry', 'Delhi', 29), (2, 'Mira', 'Mumbai', 24), (3, 'Emma', 'Bangalore', 30), (4, 'Kevin', 'Mumbai', 19)]
Exporting data using CSV file
To export data from a table into a CSV file, we use TO keyword with the COPY command.Â
Syntax: COPY <table name> TO 'location + file_name' DELIMITER ',' CSV HEADER;
Here, <table name> is the table from which we are exporting data.
'location + file_name' contains the path of the CSV file into which we want to export the data. Make sure you have 'write' access to the file.
Example:
Python3
import psycopg2
# connection establishment
conn = psycopg2.connect(
database="postgres",
user='postgres',
password='password',
host='localhost',
port='5432'
)
conn.autocommit = True
# Creating a cursor object
cursor = conn.cursor()
# query to export table into csv
sql = '''COPY demo TO
'C:\\Users\\DELL\\Downloads\\Exported_data.csv'
DELIMITER ',' CSV HEADER'''
# executing above query
cursor.execute(sql)
# Closing the connection
conn.close()
The exported CSV will look like this:
Similar Reads
Python Tutorial | Learn Python Programming Language Python Tutorial â Python is one of the most popular programming languages. Itâs simple to use, packed with features and supported by a wide range of libraries and frameworks. Its clean syntax makes it beginner-friendly.Python is:A high-level language, used in web development, data science, automatio
10 min read
Python Interview Questions and Answers Python is the most used language in top companies such as Intel, IBM, NASA, Pixar, Netflix, Facebook, JP Morgan Chase, Spotify and many more because of its simplicity and powerful libraries. To crack their Online Assessment and Interview Rounds as a Python developer, we need to master important Pyth
15+ min read
Non-linear Components In electrical circuits, Non-linear Components are electronic devices that need an external power source to operate actively. Non-Linear Components are those that are changed with respect to the voltage and current. Elements that do not follow ohm's law are called Non-linear Components. Non-linear Co
11 min read
Python OOPs Concepts Object Oriented Programming is a fundamental concept in Python, empowering developers to build modular, maintainable, and scalable applications. By understanding the core OOP principles (classes, objects, inheritance, encapsulation, polymorphism, and abstraction), programmers can leverage the full p
11 min read
Python Projects - Beginner to Advanced Python is one of the most popular programming languages due to its simplicity, versatility, and supportive community. Whether youâre a beginner eager to learn the basics or an experienced programmer looking to challenge your skills, there are countless Python projects to help you grow.Hereâs a list
10 min read
Python Exercise with Practice Questions and Solutions Python Exercise for Beginner: Practice makes perfect in everything, and this is especially true when learning Python. If you're a beginner, regularly practicing Python exercises will build your confidence and sharpen your skills. To help you improve, try these Python exercises with solutions to test
9 min read
Python Programs Practice with Python program examples is always a good choice to scale up your logical understanding and programming skills and this article will provide you with the best sets of Python code examples.The below Python section contains a wide collection of Python programming examples. These Python co
11 min read
Spring Boot Tutorial Spring Boot is a Java framework that makes it easier to create and run Java applications. It simplifies the configuration and setup process, allowing developers to focus more on writing code for their applications. This Spring Boot Tutorial is a comprehensive guide that covers both basic and advance
10 min read
Class Diagram | Unified Modeling Language (UML) A UML class diagram is a visual tool that represents the structure of a system by showing its classes, attributes, methods, and the relationships between them. It helps everyone involved in a projectâlike developers and designersâunderstand how the system is organized and how its components interact
12 min read
Enumerate() in Python enumerate() function adds a counter to each item in a list or other iterable. It turns the iterable into something we can loop through, where each item comes with its number (starting from 0 by default). We can also turn it into a list of (number, item) pairs using list().Let's look at a simple exam
3 min read