Dictionary-based CSV handling in Python - Step-by-Step Execution

Concept Flow - Dictionary-based CSV handling

Open CSV file

↓

Create DictReader

↓

Read each row as dict

↓

Process or print dict

↓

Close file

This flow shows how Python reads a CSV file using a dictionary reader, turning each row into a dictionary for easy access by column names.

Execution Sample

Python

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

Reads a CSV file and prints each row as a dictionary with column headers as keys.

Execution Table

Step	Action	Evaluation	Result
1	Open 'data.csv' file	File opened successfully	File object created
2	Create DictReader with file	DictReader ready	Reader object created
3	Read first row	Row data: 'Alice,30,NY'	{'Name': 'Alice', 'Age': '30', 'City': 'NY'}
4	Print first row dict	Output dict	{'Name': 'Alice', 'Age': '30', 'City': 'NY'}
5	Read second row	Row data: 'Bob,25,LA'	{'Name': 'Bob', 'Age': '25', 'City': 'LA'}
6	Print second row dict	Output dict	{'Name': 'Bob', 'Age': '25', 'City': 'LA'}
7	Read third row	Row data: 'Charlie,35,Chicago'	{'Name': 'Charlie', 'Age': '35', 'City': 'Chicago'}
8	Print third row dict	Output dict	{'Name': 'Charlie', 'Age': '35', 'City': 'Chicago'}
9	No more rows	Stop iteration	Exit loop and close file

💡 All rows read, iteration ends, file closed

Variable Tracker

Variable	Start	After Step 3	After Step 5	After Step 7	Final
file	None	File object	File object	File object	Closed
reader	None	DictReader object	DictReader object	DictReader object	DictReader object
row	None	{'Name': 'Alice', 'Age': '30', 'City': 'NY'}	{'Name': 'Bob', 'Age': '25', 'City': 'LA'}	{'Name': 'Charlie', 'Age': '35', 'City': 'Chicago'}	None (end)

Key Moments - 3 Insights

Why does each row come as a dictionary instead of a list?

Because DictReader uses the first CSV row as keys, so each row is a dictionary with column names as keys (see steps 3,5,7 in execution_table).

What happens if the CSV file has missing columns in some rows?

DictReader fills missing columns with None for those keys, so the dictionary still has all keys but some values may be None.

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

'with open' automatically closes the file after the block ends, ensuring no file remains open (see step 9).

Visual Quiz - 3 Questions

Look at the execution_table, what is the value of 'row' after step 5?

A {'Name': 'Alice', 'Age': '30', 'City': 'NY'}

B {'Name': 'Bob', 'Age': '25', 'City': 'LA'}

C {'Name': 'Charlie', 'Age': '35', 'City': 'Chicago'}

D None

💡 Hint

Check the 'row' variable in variable_tracker after Step 5.

At which step does the CSV reading loop stop according to the execution_table?

A Step 7

B Step 5

C Step 9

D Step 3

💡 Hint

Look for the step where 'No more rows' and 'Exit loop' are mentioned.

If the CSV file had an extra column 'Country', how would the dictionaries change?

A They would include a new key 'Country' with its values.

B They would ignore the extra column completely.

C They would raise an error when reading.

D They would only include 'Country' and ignore other columns.

💡 Hint

DictReader uses the header row as keys, so extra columns become new keys.

Concept Snapshot

Use csv.DictReader to read CSV files as dictionaries.
Each row becomes a dict with column headers as keys.
Access data by column names, not indexes.
Use 'with open' to auto-close files.
Missing columns get None values.
Easy to process CSV data by names.

Full Transcript

This visual execution shows how Python reads a CSV file using the csv.DictReader class. First, the file is opened. Then, DictReader is created with the file object. Each row is read one by one as a dictionary where keys are the column headers from the first row. The program prints each dictionary row. When no more rows remain, the loop ends and the file is closed automatically. Variables like 'row' change each iteration to hold the current dictionary. Beginners often wonder why rows are dictionaries: it's because DictReader uses the header row as keys. Also, using 'with open' ensures the file closes properly. If the CSV had extra columns, those become new keys in the dictionaries. This method makes CSV data easy to work with by column names instead of numeric indexes.

Practice

1. What is the main advantage of using csv.DictReader over csv.reader when reading CSV files?

easy

A. It writes data back to the CSV file.

B. It reads the entire file into memory at once.

C. It automatically converts all values to integers.

D. It allows accessing data by column names instead of index positions.

Solution

1. Step 1: Understand csv.reader behavior

   csv.reader reads CSV rows as lists, so you access data by index positions.

2. Step 2: Understand csv.DictReader behavior

   csv.DictReader reads rows as dictionaries, letting you access data by column names, which is clearer and safer if column order changes.

3. Final Answer:

   It allows accessing data by column names instead of index positions. -> Option D

4. Quick Check:

   DictReader uses column names for access [OK]

Hint: DictReader uses column names, not positions, for easier access [OK]

Common Mistakes:

• Thinking DictReader reads entire file at once
• Assuming DictReader converts data types automatically
• Confusing reading with writing functions

2. Which of the following is the correct way to create a csv.DictWriter object to write a CSV with columns 'name' and 'age'?

easy

A. csv.DictWriter(file, fieldnames=['name', 'age'])

B. csv.DictWriter(file, columns=['name', 'age'])

C. csv.DictWriter(file, keys=['name', 'age'])

D. csv.DictWriter(file, headers=['name', 'age'])

Solution

1. Step 1: Recall the parameter name for columns in DictWriter

   The correct parameter to specify column names is fieldnames.

2. Step 2: Check the options

   Only csv.DictWriter(file, fieldnames=['name', 'age']) uses fieldnames correctly; others use incorrect parameter names.

3. Final Answer:

   csv.DictWriter(file, fieldnames=['name', 'age']) -> Option A

4. Quick Check:

   Use fieldnames to set columns [OK]

Hint: Use 'fieldnames' to specify columns in DictWriter [OK]

Common Mistakes:

• Using 'columns' or 'keys' instead of 'fieldnames'
• Forgetting to pass a file object first
• Confusing DictReader and DictWriter parameters

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

import csv
from io import StringIO

csv_data = "name,age\nAlice,30\nBob,25"
file = StringIO(csv_data)
reader = csv.DictReader(file)
for row in reader:
    print(row['name'], row['age'])

medium

A. Alice 30
Bob 25

B. ['Alice', '30']
['Bob', '25']

C. {'name': 'Alice', 'age': '30'}
{'name': 'Bob', 'age': '25'}

D. 30 Alice
25 Bob

Solution

1. Step 1: Understand the CSV data and DictReader

   The CSV has two rows with columns 'name' and 'age'. DictReader reads each row as a dictionary.

2. Step 2: Analyze the print statement

   It prints the values of 'name' and 'age' keys separated by space for each row.

3. Final Answer:

   Alice 30 Bob 25 -> Option A

4. Quick Check:

   Prints name and age values separated by space [OK]

Hint: DictReader rows are dicts; print keys to get values [OK]

Common Mistakes:

• Printing the whole dictionary instead of values
• Mixing order of printed values
• Confusing list output with string output

4. Identify the error in this code that writes a CSV file using csv.DictWriter:

import csv
with open('output.csv', 'w') as f:
    writer = csv.DictWriter(f, fieldnames=['name', 'age'])
    writer.writerow({'name': 'Alice', 'age': 30})
    writer.writerow({'name': 'Bob', 'age': 25})

medium

A. Dictionaries passed to writerow must have string values only.

B. Fieldnames list should be a tuple, not a list.

C. Missing call to writer.writeheader() before writing rows.

D. The file should be opened in binary mode 'wb'.

Solution

1. Step 1: Check DictWriter usage

   DictWriter requires calling writeheader() to write the header row before writing data rows.

2. Step 2: Verify other parts

   Opening file in text mode 'w' is correct in Python 3, fieldnames can be a list, and values can be int or str.

3. Final Answer:

   Missing call to writer.writeheader() before writing rows. -> Option C

4. Quick Check:

   Always call writeheader() before writerow() [OK]

Hint: Call writeheader() before writing rows with DictWriter [OK]

Common Mistakes:

• Forgetting writeheader() call
• Opening file in binary mode unnecessarily
• Thinking fieldnames must be tuple
• Assuming all values must be strings

5. You have a CSV file with columns 'id', 'name', and 'score'. You want to read it using csv.DictReader and create a dictionary mapping each 'id' to the 'score' as an integer. Which code snippet correctly does this?

hard

A. with open('data.csv') as f:
reader = csv.DictReader(f)
result = {int(row['id']): row['score'] for row in reader}

B. with open('data.csv') as f:
reader = csv.DictReader(f)
result = {row['id']: int(row['score']) for row in reader}

C. with open('data.csv') as f:
reader = csv.reader(f)
result = {row['id']: int(row['score']) for row in reader}

D. with open('data.csv') as f:
reader = csv.DictReader(f)
result = {row['score']: int(row['id']) for row in reader}

Solution

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

   Only csv.DictReader allows accessing 'id' and 'score' by keys.

2. Step 2: Create dictionary with 'id' as key and integer 'score' as value

   with open('data.csv') as f: reader = csv.DictReader(f) result = {row['id']: int(row['score']) for row in reader} correctly converts 'score' to int and uses 'id' as key.

3. Final Answer:

   with open('data.csv') as f: reader = csv.DictReader(f) result = {row['id']: int(row['score']) for row in reader} -> Option B

4. Quick Check:

   DictReader + dict comprehension + int conversion [OK]

Hint: Use DictReader and dict comprehension with int() conversion [OK]

Common Mistakes:

• Using csv.reader instead of DictReader
• Swapping keys and values in dictionary
• Not converting score to int
• Converting id to int instead of score