Working with CSV files in Python - Step-by-Step Execution

Concept Flow - Working with CSV files

Open CSV file

↓

Create CSV reader

↓

Read each row

↓

Process row data

↓

Close file

This flow shows how Python opens a CSV file, reads it row by row, processes the data, and then closes the file.

Execution Sample

Python

import csv
with open('data.csv', newline='') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

This code opens a CSV file named 'data.csv', reads each row, and prints it as a list.

Execution Table

Step	Action	Code Line	Variable State	Output
1	Open file 'data.csv' for reading	with open('data.csv', newline='') as file:	file=open
2	Create CSV reader object	reader = csv.reader(file)	reader=csv.reader object
3	Read first row	for row in reader:	row=['Name', 'Age', 'City']	['Name', 'Age', 'City']
4	Print first row	print(row)	row=['Name', 'Age', 'City']	['Name', 'Age', 'City']
5	Read second row	for row in reader:	row=['Alice', '30', 'New York']	['Alice', '30', 'New York']
6	Print second row	print(row)	row=['Alice', '30', 'New York']	['Alice', '30', 'New York']
7	Read third row	for row in reader:	row=['Bob', '25', 'Los Angeles']	['Bob', '25', 'Los Angeles']
8	Print third row	print(row)	row=['Bob', '25', 'Los Angeles']	['Bob', '25', 'Los Angeles']
9	No more rows	for row in reader:	row=None
10	Close file automatically	end of with block	file closed

💡 No more rows to read, loop ends and file closes automatically

Variable Tracker

Variable	Start	After Step 3	After Step 5	After Step 7	Final
file	None	open	open	open	closed
reader	None	csv.reader object	csv.reader object	csv.reader object	csv.reader object
row	None	['Name', 'Age', 'City']	['Alice', '30', 'New York']	['Bob', '25', 'Los Angeles']	None

Key Moments - 3 Insights

Why does the 'row' variable change each loop iteration?

Because the 'for' loop reads the next line from the CSV file each time, updating 'row' with the new list of values (see steps 3, 5, 7 in execution_table).

What happens when there are no more rows to read?

The loop ends because the CSV reader returns no more data, so the loop stops (step 9). The file then closes automatically (step 10).

Why do we use 'with open' instead of just 'open'?

'with open' ensures the file closes automatically after the block finishes, even if errors happen, so we don't have to close it manually (step 10).

Visual Quiz - 3 Questions

Look at the execution table, what is the value of 'row' at step 5?

A ['Alice', '30', 'New York']

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

C ['Bob', '25', 'Los Angeles']

D None

💡 Hint

Check the 'Variable State' column at step 5 in the execution_table.

At which step does the file close automatically?

A Step 9

B Step 10

C Step 3

D Step 1

💡 Hint

Look for the step mentioning 'Close file automatically' in the execution_table.

If the CSV file had 5 rows instead of 3, how would the variable 'row' change?

A It would be None from the start

B It would stay the same as in the example

C It would have 5 different lists before becoming None

D It would cause an error

💡 Hint

Refer to the variable_tracker showing 'row' changing each iteration for each CSV row.

Concept Snapshot

Working with CSV files in Python:
- Use 'import csv' to access CSV tools.
- Open file with 'with open(filename) as file:' for safe handling.
- Create a reader with 'csv.reader(file)'.
- Loop over reader to get each row as a list.
- File closes automatically after 'with' block ends.

Full Transcript

This lesson shows how Python reads CSV files step-by-step. First, the program opens the file safely using 'with open'. Then it creates a CSV reader object to read rows. Each row is a list of strings representing the columns. The program loops over each row, printing it. When no rows remain, the loop ends and the file closes automatically. Variables like 'row' update each loop to hold the current row's data. Using 'with open' ensures the file closes even if errors happen. This method is simple and safe for reading CSV files.

Practice

1. What does the Python csv.reader function do when working with CSV files?

easy

A. Reads the CSV file and returns each row as a list of values

B. Writes data to a CSV file

C. Deletes a CSV file

D. Converts CSV data into JSON format

Solution

1. Step 1: Understand the purpose of csv.reader

   The csv.reader function reads CSV files and returns each row as a list of strings representing the columns.

2. Step 2: Differentiate from other CSV functions

   Functions like writing or deleting files are not done by csv.reader. It only reads and parses rows.

3. Final Answer:

   Reads the CSV file and returns each row as a list of values -> Option A

4. Quick Check:

   csv.reader reads rows as lists [OK]

Hint: Remember: reader reads rows as lists [OK]

Common Mistakes:

• Confusing reader with writer
• Thinking it deletes files
• Assuming it converts formats

2. Which of the following is the correct way to open a CSV file for reading in Python?

easy

A. open('data.csv', 'a')

B. open('data.csv', 'w')

C. open('data.csv', 'r')

D. open('data.csv', 'x')

Solution

1. Step 1: Understand file modes in Python

   The mode 'r' means open for reading, which is needed to read a CSV file.

2. Step 2: Check other modes

   'w' is for writing (overwrites), 'a' is for appending, and 'x' is for creating a new file. None are for reading existing files.

3. Final Answer:

   open('data.csv', 'r') -> Option C

4. Quick Check:

   Use 'r' mode to read files [OK]

Hint: Use 'r' mode to read files [OK]

Common Mistakes:

• Using 'w' which overwrites file
• Using 'a' which appends data
• Using 'x' which fails if file exists

3. What will be the output of this code snippet?

import csv
with open('data.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(['Name', 'Age'])
    writer.writerow(['Alice', '30'])

with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    rows = list(reader)
print(rows)

medium

A. ['Name', 'Age', 'Alice', '30']

B. SyntaxError

C. [['Name, Age'], ['Alice, 30']]

D. [['Name', 'Age'], ['Alice', '30']]

Solution

1. Step 1: Writing rows with csv.writer

   The code writes two rows: header ['Name', 'Age'] and data ['Alice', '30'] as lists.

2. Step 2: Reading rows with csv.reader

   Reading back returns a list of lists, each inner list is a row split by commas.

3. Final Answer:

   [['Name', 'Age'], ['Alice', '30']] -> Option D

4. Quick Check:

   csv.reader returns list of lists [OK]

Hint: csv.reader returns list of lists, not flat list [OK]

Common Mistakes:

• Expecting a flat list instead of list of lists
• Thinking rows are single strings
• Syntax errors from missing newline='' in open

4. Identify the error in this code that reads a CSV file:

import csv
with open('data.csv', 'r') as f:
    reader = csv.reader(f)
    for row in reader:
    print(row)

medium

A. csv.reader cannot be used with 'with' statement

B. Indentation error in the for loop body

C. File mode should be 'w' instead of 'r'

D. Missing import statement

Solution

1. Step 1: Check indentation inside the for loop

   The print statement must be indented inside the for loop to run for each row.

2. Step 2: Verify other parts

   Import is present, file mode 'r' is correct for reading, and csv.reader works with 'with' statement.

3. Final Answer:

   Indentation error in the for loop body -> Option B

4. Quick Check:

   Indent loop body correctly [OK]

Hint: Indent inside loops to avoid errors [OK]

Common Mistakes:

• Not indenting loop body
• Changing file mode incorrectly
• Thinking csv.reader can't be used with 'with'

5. You have a CSV file with columns 'Name', 'Age', and 'City'. You want to read it and create a dictionary where keys are names and values are ages (as integers). Which code snippet correctly does this?

hard

A. import csv
with open('data.csv', 'r') as f:
reader = csv.DictReader(f)
result = {row['Name']: int(row['Age']) for row in reader}
print(result)

B. import csv
with open('data.csv', 'r') as f:
reader = csv.reader(f)
result = {row[0]: int(row[1]) for row in reader}
print(result)

C. import csv
with open('data.csv', 'r') as f:
reader = csv.DictReader(f)
result = {row['Age']: row['Name'] for row in reader}
print(result)

D. import csv
with open('data.csv', 'r') as f:
reader = csv.reader(f)
result = {int(row[1]): row[0] for row in reader}
print(result)

Solution

1. Step 1: Use csv.DictReader to access columns by name

   DictReader reads rows as dictionaries, so we can use keys like 'Name' and 'Age'.

2. Step 2: Create dictionary with names as keys and ages as integer values

   The comprehension uses row['Name'] as key and converts row['Age'] to int for value.

3. Final Answer:

   import csv with open('data.csv', 'r') as f: reader = csv.DictReader(f) result = {row['Name']: int(row['Age']) for row in reader} print(result) -> Option A

4. Quick Check:

   DictReader + dict comprehension with int conversion [OK]

Hint: Use DictReader and convert age to int in comprehension [OK]

Common Mistakes:

• Using csv.reader without column names
• Swapping keys and values
• Not converting age to int