How to read csv in Python?

Reading CSV Files in Python: A Comprehensive Guide

Introduction

CSV (Comma Separated Values) is a widely used format for storing tabular data in a plain text file. Python provides an efficient way to read and manipulate CSV files using the built-in csv module. In this article, we will explore the steps to read CSV files in Python, including how to handle errors, specify column names, and perform data manipulation.

Importing the csv Module

Before we can read a CSV file, we need to import the csv module. We can do this by adding the following line at the top of our Python script:

import csv

Reading a CSV File

To read a CSV file, we can use the csv.reader object, which returns an iterator over the rows of the file. Here’s an example:

# Open the CSV file in read mode

with open('example.csv', 'r') as file:

    # Create a csv.reader object

    reader = csv.reader(file)

    # Iterate over the rows

    for row in reader:

        # Print the row

        print(row)

This will output:

['Name', 'Age', 'City']

['John', '25', 'New York']

['Alice', '30', 'Los Angeles']

Specifying Column Names

When reading a CSV file, we can specify the column names using the csv.reader object. We can do this by passing a list of column names to the fieldnames parameter:

# Open the CSV file in read mode

with open('example.csv', 'r') as file:

    # Create a csv.reader object

    reader = csv.reader(file)

    # Specify the column names

    fieldnames = ['Name', 'Age', 'City']

    # Iterate over the rows

    for row in reader:

        # Print the row

        print(row)

This will output:

['Name', 'Age', 'City']

['John', '25', 'New York']

['Alice', '30', 'Los Angeles']

Handling Errors

When reading a CSV file, we can handle errors using try-except blocks. Here’s an example:

# Open the CSV file in read mode

with open('example.csv', 'r') as file:

    try:

        # Create a csv.reader object

        reader = csv.reader(file)

        # Iterate over the rows

        for row in reader:

            # Print the row

            print(row)

    except csv.Error as e:

        # Handle the error

        print(f"Error reading CSV file: {e}")

This will output:

Error reading CSV file: csv.Error: invalid literal for int() with base 10: 'Name'

Specifying the Encoding

When reading a CSV file, we can specify the encoding using the encoding parameter. Here’s an example:

# Open the CSV file in read mode with encoding

with open('example.csv', 'r', encoding='utf-8') as file:

    # Create a csv.reader object

    reader = csv.reader(file)

    # Iterate over the rows

    for row in reader:

        # Print the row

        print(row)

This will output:

['Name', 'Age', 'City']

['John', '25', 'New York']

['Alice', '30', 'Los Angeles']

Performing Data Manipulation

Once we have read a CSV file, we can perform data manipulation using various methods. Here’s an example:

# Open the CSV file in read mode

with open('example.csv', 'r') as file:

    # Create a csv.reader object

    reader = csv.reader(file)

    # Iterate over the rows

    for row in reader:

        # Print the row

        print(row)

        # Extract the values

        name = row[0]

        age = row[1]

        city = row[2]

        # Print the values

        print(f"Name: {name}, Age: {age}, City: {city}")

This will output:

Name: John, Age: 25, City: New York

Name: Alice, Age: 30, City: Los Angeles

Conclusion

Reading CSV files in Python is a straightforward process that can be accomplished using the csv module. By specifying column names, handling errors, and performing data manipulation, we can efficiently read and manipulate CSV files in Python.

Table of Contents