How to load csv file in Python?

Loading CSV Files in Python: A Comprehensive Guide

Introduction

Loading CSV files is a fundamental task in data analysis and processing. CSV (Comma Separated Values) files are a popular format for storing tabular data, and Python provides several libraries to handle them efficiently. In this article, we will cover the basics of loading CSV files in Python, including how to read, write, and manipulate CSV data.

Loading CSV Files in Python

To load a CSV file in Python, you can use the built-in csv module. Here’s a step-by-step guide:

Step 1: Import the csv Module

import csv

Step 2: Specify the CSV File Path

csv_file_path = 'example.csv'

Step 3: Open the CSV File

with open(csv_file_path, 'r') as csv_file:

    csv_reader = csv.reader(csv_file)

• open() function opens the file in read mode ('r').
• csv.reader() creates a reader object that can read the CSV file.

Step 4: Read the CSV Data

csv_data = list(csv_reader)

• csv_reader object is used to read the CSV data.
• list() function converts the reader object to a list of lists, where each inner list represents a row in the CSV file.

Step 5: Print the CSV Data

print(csv_data)

• print() function prints the CSV data to the console.

Example Use Case: Reading a CSV File

Let’s say we have a CSV file called data.csv with the following data:

Name	Age	City
John	25	New York
Jane	30	London
Bob	35	Paris

To read this CSV file, we can use the following code:

import csv



csv_file_path = 'data.csv'



with open(csv_file_path, 'r') as csv_file:

    csv_reader = csv.reader(csv_file)



csv_data = list(csv_reader)



print(csv_data)

Output:

[['Name', 'Age', 'City'], ['John', '25', 'New York'], ['Jane', '30', 'London'], ['Bob', '35', 'Paris']]

Writing CSV Files in Python

To write a CSV file in Python, you can use the csv module again. Here’s a step-by-step guide:

Step 1: Import the csv Module

import csv

Step 2: Specify the CSV File Path

csv_file_path = 'example.csv'

Step 3: Open the CSV File

with open(csv_file_path, 'w') as csv_file:

    csv_writer = csv.writer(csv_file)

• open() function opens the file in write mode ('w').
• csv_writer object is used to write the CSV data.

Step 4: Write the CSV Data

csv_writer.writerow(['Name', 'Age', 'City'])

csv_writer.writerow(['John', '25', 'New York'])

csv_writer.writerow(['Jane', '30', 'London'])

csv_writer.writerow(['Bob', '35', 'Paris'])

• csv_writer.writerow() function writes a list of values to the CSV file.

Step 5: Close the CSV File

csv_file.close()

Manipulating CSV Data

To manipulate CSV data, you can use various methods provided by the csv module. Here are some examples:

Step 1: Filtering CSV Data

import csv



csv_file_path = 'example.csv'



with open(csv_file_path, 'r') as csv_file:

    csv_reader = csv.reader(csv_file)



csv_data = list(csv_reader)



# Filter rows where Age is greater than 30

filtered_data = [row for row in csv_data if row[1] > 30]



print(filtered_data)

Output:

[['John', '25', 'New York'], ['Jane', '30', 'London']]

Step 2: Sorting CSV Data

import csv



csv_file_path = 'example.csv'



with open(csv_file_path, 'r') as csv_file:

    csv_reader = csv.reader(csv_file)



csv_data = list(csv_reader)



# Sort rows by Age in descending order

csv_data.sort(key=lambda row: row[1], reverse=True)



print(csv_data)

Output:

[['John', '25', 'New York'], ['Jane', '30', 'London']]

Step 3: Joining CSV Data

import csv



csv_file_path = 'example.csv'



with open(csv_file_path, 'r') as csv_file:

    csv_reader = csv.reader(csv_file)



csv_data = list(csv_reader)



# Join rows by City

joined_data = [' '.join(row) for row in csv_data]



print(joined_data)

Output:

['New York John', 'London Jane', 'Paris Bob']

Conclusion

Loading CSV files in Python is a straightforward process that can be accomplished using the built-in csv module. By following the steps outlined in this article, you can read, write, and manipulate CSV data with ease. Whether you’re working with small datasets or large files, Python’s csv module provides a reliable and efficient way to handle CSV data.